简介
在这个挑战中,你将编写一个 Python 函数,该函数根据给定过滤函数的结果将列表拆分为两组。这是编程中的一项常见任务,尤其是在处理大型数据集时。
根据函数对列表进行二分
编写一个函数 bifurcate_by(lst, fn),它接受一个列表 lst 和一个过滤函数 fn 作为参数。该函数应根据过滤函数的结果将列表拆分为两组。如果过滤函数对某个元素返回真值,则应将其添加到第一组。否则,应将其添加到第二组。
你的函数应返回一个包含两个列表的列表,其中第一个列表包含过滤函数返回真值的所有元素,第二个列表包含过滤函数返回假值的所有元素。
使用列表推导式根据 fn 对每个元素返回的值将元素添加到相应的组中。
def bifurcate_by(lst, fn):
return [
[x for x in lst if fn(x)],
[x for x in lst if not fn(x)]
]
bifurcate_by(['beep', 'boop', 'foo', 'bar'], lambda x: x[0] == 'b')
## [ ['beep', 'boop', 'bar'], ['foo'] ]
总结
在这个挑战中,你学习了如何根据给定过滤函数的结果将列表拆分为两组。你使用列表推导式,根据过滤函数对每个元素返回的值将元素添加到相应的组中。这是编程中的一项实用技术,尤其是在处理大型数据集时。