python - 选择具有给定属性值的网络节点

我想选择图形中具有特定属性的节点并对其执行操作。 如何选择具有给定属性值的节点？例如:

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

有没有办法只选择 at == 5 的节点？

我在想象类似的东西(这行不通):

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

最佳答案

Python <= 2.7:

根据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)

Python 2.7 和 3:

nodesAt5 = [x for x,y in P.nodes(data=True) if y['at']==5]

关于python - 选择具有给定属性值的网络节点，我们在Stack Overflow上找到一个类似的问题： https://stackoverflow.com/questions/31922917/