Выберите узлы сети с заданным значением атрибута

Question

Я хотел бы выбрать и выполнить операции над узлами в графе с определенными атрибутами. Как выбрать узлы с заданным значением атрибута? Например:

P=nx.Graph() 
P.add_node('node1',at=5) 
P.add_node('node2',at=5) 
P.add_node('node3',at=6) 

Есть ли способ, чтобы выбрать только те узлы, с в == 5 ?.

я себе что-то вроде (это не работает):

for p in P.nodes(): 
    P.node[p]['at'==5] 

Answer 1

Согласно documentation попробовать:

nodesAt5 = filter(lambda (n, d): d['at'] == 5, P.nodes(data=True)) 

или как ваш подход

nodesAt5 = [] 
for (p, d) in P.nodes(data=True): 
    if d['at'] == 5: 
     nodesAt5.append(p)