根据给定的属性值选择网络节点

Question

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

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]

- Lee

Answer 1

Python <= 2.7:

根据文档，请尝试：

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]

- wenzul

---

谢谢，根据您的回答，我使用了这个：[x for x,y in P.nodes(data=True) if y['at']==5]。 - Lee

@atomh33ls 很好。是的，使用列表推导也是可能的。如果您将使用大量节点，则可能更喜欢 filter。它应该会快一点。 - wenzul

请注意，在Python 3中，lambda (n, d) 元组解包语法已不再起作用。 - amoe

Answer 2

如果有人卡住了，以下是示例代码：

import networkx as nx

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

# You can select like this
selected_data = dict( (n,d['at']) for n,d in P.nodes().items() if d['at'] == 5)
# Then do what you want to do with selected_data
print(f'Node found : {len (selected_data)} : {selected_data}')

- Abhi