在networkx中查找满足某个标准的边

Question

我有一个在节点上有属性的networkx digraph，我想找出特定节点属性不同的所有边。有没有一种方法可以自动做到这一点，还是必须用edge_iter(data=True)迭代并自己找到它们？在

Answer 1

此功能没有内置函数，但非常简单：

import networkx as nx

G = nx.Graph()
G.add_node(1,color='red')
G.add_node(2,color='red')
G.add_node(3,color='blue')
G.add_node(4,color='blue')

G.add_edges_from([(1,2),(1,3),(3,4)])

for (u,v) in G.edges_iter():
    if G.node[u]['color'] != G.node[v]['color']:
        print u,v