在一定条件下从列表(或其他可迭代的)中获取项目数
问题描述
假设我有一个包含大量项目的列表,
l = [ 1, 4, 6, 30, 2, ... ]
我要获取该列表中的项目数,其中一个项目满足特定条件。我的第一个想法是:
count = len([i for i in l if my_condition(l)])
但是,如果过滤后的列表也有大量的项目,我认为
为过滤结果创建新列表只是浪费内存。为了提高效率,IMHO,上面的调用不能优于:
count = 0
for i in l:
if my_condition(l):
count += 1
有没有什么函数式方法可以在不生成临时列表的情况下获取满足条件的项目数?
解决方案
您可以使用generator expression:
>>> l = [1, 3, 7, 2, 6, 8, 10]
>>> sum(1 for i in l if i % 4 == 3)
2
甚至
>>> sum(i % 4 == 3 for i in l)
2
它使用的事实是True == 1
和False == 0
。
或者,您也可以使用itertools.imap
(Python2)或简单的map
(Python3):
>>> def my_condition(x):
... return x % 4 == 3
...
>>> sum(map(my_condition, l))
2
相关文章