python - 如何在 Networkx Python 中添加带有列表作为标签的边

Question

我有这个例子:

G.add_nodes_from(List1)
G.add_nodes_from(List2)
edgeList = list(zip(List1, List2))
for item in List3:
    G.add_edges_from(edgeList, label = item) 

我想要的结果是两个列表之间的每条边都有列表3的相应元素作为标签。像这样的事情:

(List1[0], List2[0], {'label': List3[0]})

Answer 1

注意:我已将变量名称从 List1 更改为 source_nodes，将 List2 更改为 dest_nodes 和 List3 到以下代码中的 data_nodes。

您可以尝试在所有三个列表上使用 zip 并使用 nx.add_edge 添加边缘。 。像这样的事情:

import networkx as nx

source_nodes = ['A', 'B', 'C', 'D']
dest_nodes = ['P', 'Q', 'R', 'S']
data_nodes = ['W', 'X', 'Y', 'Z']

G = nx.DiGraph()

# Each element of this zip will be
# (source[i], dest[i], data[i]) 
for u,v,d in zip(source_nodes, dest_nodes, data_nodes):
    G.add_edge(u, v, label=d)

print(G.edges(data=True))
# OutEdgeDataView([('A', 'P', {'label': 'W'}),
#                  ('B', 'Q', {'label': 'X'}),
#                  ('C', 'R', {'label': 'Y'}),
#                  ('D', 'S', {'label': 'Z'})])

您还可以在 this Google Colab Notebook 中查看代码了解更多信息。