Python:获取图的边缘端点

时间:2010-01-28 04:35:57

标签: python

我需要图表中的边缘端点。我安装了networkx。我知道如何继续。

networkx.Graph.edges_iter()返回图中的所有边

[e for e in G.edges_iter()]
[(0, 1), (1, 2), (2, 3)]

我想要的是一个列表[0,1,1,2,3]

如何从上述数据中获得此信息?

2 个答案:

答案 0 :(得分:2)

>>> import itertools
>>> list(itertools.chain(*[(0, 1), (1, 2), (2, 3)]))
[0, 1, 1, 2, 2, 3]

您可能不需要list(...),因为它已经可以迭代。

您也可以直接尝试itertools.chain(G.edges_iter())

答案 1 :(得分:1)

edgeList = []
for (a,b) in G.edges_iter():
    edgeList.append(a)
    edgeList.append(b)