在一定条件下从列表(或其他可迭代的)中获取项目数

Get number of items from list (or other iterable) with certain condition(在一定条件下从列表(或其他可迭代的)中获取项目数)

问题描述

假设我有一个包含大量项目的列表，

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

这篇关于在一定条件下从列表(或其他可迭代的)中获取项目数的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持编程学习网！