Python代码阅读(第24篇):根据筛选函数将列表元素分成两组

x33g5p2x  于2021-10-28 转载在 Python  
字(0.5k)|赞(0)|评价(0)|浏览(118)

本篇阅读的代码实现根据筛选函数将列表元素分成两组。 本篇阅读的代码片段来自于30-seconds-of-python。

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时,归入第二组。

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

相关文章