根据值将列表分成两组
根据给定的 filter
列表的结果,将值分成两组。
- 使用列表推导式和
zip()
将元素添加到组中,根据filter
的值。 - 如果
filter
对于任何元素都为真值,则将其添加到第一组,否则将其添加到第二组。
def bifurcate(lst, filter):
return [
[x for x, flag in zip(lst, filter) if flag],
[x for x, flag in zip(lst, filter) if not flag]
]
bifurcate(['beep', 'boop', 'foo', 'bar'], [True, True, False, True])
# [ ['beep', 'boop', 'bar'], ['foo'] ]