具有匹配属性的对象列表中的Python计数元素

Question

我试图找到一种简单快速的方法来计算列表中符合条件的对象数量.例如

class Person:
    def __init__(self, Name, Age, Gender):
        self.Name = Name
        self.Age = Age
        self.Gender = Gender

# List of People
PeopleList = [Person("Joan", 15, "F"), 
              Person("Henry", 18, "M"), 
              Person("Marg", 21, "F")]

现在,计算此列表中与基于其属性的参数匹配的对象数量的最简单函数是什么？例如,为Person.Gender =="F"或P​​erson.Age <20返回2.

Answer 1

class Person:
    def __init__(self, Name, Age, Gender):
        self.Name = Name
        self.Age = Age
        self.Gender = Gender


>>> PeopleList = [Person("Joan", 15, "F"), 
              Person("Henry", 18, "M"), 
              Person("Marg", 21, "F")]
>>> sum(p.Gender == "F" for p in PeopleList)
2
>>> sum(p.Age < 20 for p in PeopleList)
2

Answer 2

我知道这是一个老问题,但是现在有一种stdlib方法可以做到这一点

from collections import Counter

c = Counter(getattr(person, 'gender') for person in PeopleList)
# c now is a map of attribute values to counts -- eg: c['F']

Answer 3

我更喜欢这个：

\n\n

def count(iterable):\n    return sum(1 for _ in iterable)\n

\n\n

然后你可以像这样使用它：

\n\n

femaleCount = count(p for p in PeopleList if p.Gender == "F")\n

\n\n

这是便宜的（不会创建无用的列表等）并且完全可读（我想说比两者都好sum(1 for \xe2\x80\xa6 if \xe2\x80\xa6)）sum(p.Gender == "F" for \xe2\x80\xa6)。

\n

Answer 4

我发现使用列表理解并获得其长度比使用更快sum().

根据我的测试 ......

len([p for p in PeopleList if p.Gender == 'F'])

...运行速度是...的1.59倍

sum(p.Gender == "F" for p in PeopleList)