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

原创于 2021-10-27 15:23:37 发布 · 227 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#python #开发语言 #后端

Python代码阅读专栏收录该内容

50 篇文章

订阅专栏

本文介绍了Python代码片段，展示了如何使用列表推导式根据筛选函数将列表元素分成两组。bifurcate_by函数接受一个列表和一个筛选函数，返回两个子列表，分别包含满足和不满足筛选条件的元素。这对于初学者理解函数和列表操作很有帮助。

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

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