在networkx中找到符合特定条件的边

时间:2015-01-27 00:24:27

标签: python graph networkx

我在节点上有一个networkx digraph属性,我想找到特定节点属性不同的所有边。有没有办法自动执行此操作,还是必须使用edge_iter(data=True)进行迭代并自行查找?

1 个答案:

答案 0 :(得分:2)

没有内置功能,但很简单:

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