python - 最短的计数方式？

Question

假设我有一个 MyClass 列表。我想计算将 MyClass.SomeProperty 设置为 True 的元素的数量(假设 SomeProperty 是一个始终为 True 的 bool 值 或 False)。我的顾虑是:

• 它应该运行得很快
• 代码不应该让人难以阅读
• 它应该适用于任何类型的条件(不仅仅是 bool 值)

我知道我能做到:

count = len([e for e in MyList if e.SomeProperty])  # For non booleans, something like e.SomeProperty == MyValue

但似乎效率不高。

• 为什么要输入像 e for e 这样毫无意义的东西？
• 如果您只想计算它们，为什么还要创建一个完整的列表？

有没有可能做得更好？

Answer 1

您可以将 sum 与生成器表达式一起使用。

count = sum( e.SomeProperty for e in MyList )

或者对于一般谓词 p:

count = sum( p(e) for e in MyList )

这利用了 True 和 False 可以用作整数 1 和 0 的事实，并且使用生成器这一事实将阻止创建新列表。

如果你坚持要避免 for e in 部分，你可以使用 map 和 attrgetter :

import opertor
count = sum(map(operator.attrgetter('SomeProperty'), MyList))

或者对于一般谓词 p:

count = sum(map(p, MyList))

但是，这不是 pythonic。我推荐第一种方法。