Как проверить является ли определенный элемент в списке списков подходящим?

Question

[kolomiec_artiom]

Добрый день!

У меня есть структура следующего вида: (написанная на языке python)

[{
"member": 0,
"can_invite": 1,
"user_id": 1
}, {
"member": 0,
"can_invite": 1,
"user_id": 2
}, {
"member": 1,
"can_invite": 0,
"user_id": 3
}]

Как мне оперативно проверить сколько словарей содержат "can_invite": 1 ?

Есть идея создать список со всеми can_invite, которые равны 1 и проверить длину этого списка. Можно ли быстрее?

Answer 1

[Вадим Шаталов]

l = [{          
 "member": 0,    
 "can_invite": 1,
 "user_id": 1    
 }, {            
 "member": 0,    
 "can_invite": 1,
 "user_id": 2    
 }, {            
 "member": 1,    
 "can_invite": 0,
 "user_id": 3    
 }]              
In [3]: sum(map(lambda x: x.get('can_invite',0) == 1, l))
Out[3]: 2

Comments

[JaxxDexx]

либо:
sum(x.get("can_invite", 0) == 1 for x in l)

либо можно:
sum(x.get("can_invite", 0) for x in l)

либо, при условии что can_invite может быть только 0 и 1, и аттрибут всегда существует
sum(x["can_invite"] for x in l)

[kolomiec_artiom]

JaxxDexx, спасибо вам большое

[kolomiec_artiom]

Вадим Шаталов, спасибо вам большое

[longclaps]

kolomiec_artiom, знаешь, чего не знает типичный Python developer? А я тебе скажу - он не знает стандартной библиотеки и всюду суёт самопальные лямбды. Не будь таким, читай документацию.

from operator import itemgetter

l = [{
    "member": 0,
    "can_invite": 1,
    "user_id": 1
}, {
    "member": 0,
    "can_invite": 1,
    "user_id": 2
}, {
    "member": 1,
    "can_invite": 0,
    "user_id": 3
}]
print(sum(map(itemgetter('can_invite'), l)))