Elementi di conteggio Python in un elenco di oggetti con attributi corrispondenti

Question

Sto cercando di trovare un modo semplice e veloce per contare il numero di oggetti in un elenco che corrisponde a un criterio. ad es.

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")] 

Ora che cosa è la funzione più semplice per contare il numero di oggetti in questo elenco che corrispondono un argomento in base alle loro caratteristiche? esempio, tornando 2 per Person.Gender == "F" o Person.Age < 20.

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

Personalmente ritengo che definisce una funzione è più semplice su molteplici usi:

def count(seq, pred): 
    return sum(1 for v in seq if pred(v)) 

print(count(PeopleList, lambda p: p.Gender == "F")) 
print(count(PeopleList, lambda p: p.Age < 20)) 

Particolarmente

se vuoi riutilizzare una query.

Answer 3

preferisco questo:

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

quindi è possibile utilizzare in questo modo:

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

che è a buon mercato (non creare elenchi inutili ecc) e perfettamente leggibile (direi meglio di entrambi i modelli sum(1 for … if …) e sum(p.Gender == "F" for …)).

Answer 4

Ho scoperto che l'utilizzo di una lista di comprensione e la sua lunghezza era più veloce rispetto all'utilizzo di sum().

Secondo my tests ...

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

... corre 1,59 volte più veloce ...

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

Answer 5

So che questa è una vecchia questione, ma in questi giorni un modo per stdlib fare questo sarebbe

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']