python - 使用 Python 列表推导计算列表中的正整数元素

Question

我有一个整数列表，我需要计算其中有多少 > 0。
我目前正在使用如下所示的列表理解:

sum([1 for x in frequencies if x > 0])

这似乎是一个不错的理解，但我不太喜欢“1”；这似乎是一个神奇的数字。有没有更 Pythonish 的方式来做到这一点？

Answer 1

如果要减少内存量，可以避免使用生成器生成临时列表:

sum(x > 0 for x in frequencies)

这是因为 bool 是 int 的子类:

>>> isinstance(True,int)
True

而True的值为1:

>>> True==1
True

但是，正如 Joe Golton 在评论中指出的那样，这种解决方案并不是很快。如果您有足够的内存来使用中间临时列表，那么 sth's solution可能更快。以下是比较各种解决方案的一些时间安排:

>>> frequencies = [random.randint(0,2) for i in range(10**5)]

>>> %timeit len([x for x in frequencies if x > 0])   # sth
100 loops, best of 3: 3.93 ms per loop

>>> %timeit sum([1 for x in frequencies if x > 0])
100 loops, best of 3: 4.45 ms per loop

>>> %timeit sum(1 for x in frequencies if x > 0)
100 loops, best of 3: 6.17 ms per loop

>>> %timeit sum(x > 0 for x in frequencies)
100 loops, best of 3: 8.57 ms per loop

请注意，timeit 结果可能会因 Python、操作系统或硬件的版本而异。

当然，如果您要对大量数字进行数学运算，您可能应该使用 NumPy:

>>> frequencies = np.random.randint(3, size=10**5)
>>> %timeit (frequencies > 0).sum()
1000 loops, best of 3: 669 us per loop

NumPy 数组比等效的 Python 列表需要更少的内存，并且计算的执行速度比任何纯 Python 解决方案都要快。