根据值将列表分成两组

根据给定的 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'] ]