Python: получение конечных точек ребра графа

Question

Мне нужны конечные точки ребер на графике. Я установил networkx. У меня есть идея, как поступить.

networkx.Graph.edges_iter () возвращает все ребра графа

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

То, что я хочу, это список [0,1,1,2,2,3]

Как я могу получить это из приведенных выше данных?

Answer 1

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

Возможно, вам не понадобится list(...), потому что он уже повторяется.

И вы также можете попробовать itertools.chain(G.edges_iter()) напрямую

Answer 2

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