Python代码阅读（第24篇）：根据筛选函数将列表元素分成两组

Python 代码阅读合集介绍： 为什么不推荐Python初学者直接看项目源码

本篇阅读的代码实现根据筛选函数将列表元素分成两组。

本篇阅读的代码片段来自于30-seconds-of-python。

bifurcate_by

def bifurcate_by(lst, fn):
  return [
    [x for x in lst if fn(x)],
    [x for x in lst if not fn(x)]
  ]

# EXAMPLES
bifurcate_by(
  ['beep', 'boop', 'foo', 'bar'], 
  lambda x: x[0] == 'b'
) # [ ['beep', 'boop', 'bar'], ['foo'] ]

bifurcate_by函数接收一个列表和一个筛选函数作为输入。根据筛选函数返回的值将列表元素归入不同的组中。当筛选函数返回True时，归入第一组，当筛选函数返回False时，归入第二组。

函数使用两个列表推导式生成两个子列表，用于放置两个组的元素。