如何将边的二部列表转换为 NetworkX 中的图，每组节点都具有重叠标签？

Question

我目前正在尝试使用 NetworkX 将我拥有的 JSON 文件转换为二部图。 我的数据如下所示：

    'edges': [
   {'A': 8, 'B': 6},
   {'A': 21, 'B': 4},
   {'A': 27, 'B': 6},
   {'A': 4, 'B': 5},
   ...
   ]

基本上，我有两组节点：A 和 B。边列表描述了集合 A 的节点如何与集合 B 的节点连接。

我的问题是节点具有重叠的标识符。 换句话说，我在集合 A 中有节点 0、1、2 等，在 B 中有 0、1、2 等（节点数不一定相同，但它们的值重叠）。 当我尝试将所有节点添加到我的图中时，我没有得到所有节点，因为有些节点是重复的。

我如何处理在我的两个集合中有重复节点？ 它们实际上是不同的节点，但 NetworkX 并没有这样对待它们。 我看到您可以为节点设置“值”，但我不确定如何将所有内容组合在一起。

最后，我想有一种方法来生成节点，以便我可以通过以下方式将边添加到我的图中：

import networkx as nx

G = nx.Graph()
# Add nodes here

# Add edges
G.add_edges_from(
    (edge['B'], edge['A'])
    for edge in graph['edges']
)

Answer 1

您可以将节点添加到图形中，其 ID 不等于您的字典中的值，但等于 AB 类型加值：

import networkx as nx

# Create edges dict and graph
edges = [
   {'A': 8, 'B': 6},
   {'A': 21, 'B': 4},
   {'A': 27, 'B': 6},
   {'A': 4, 'B': 5},
]
G = nx.Graph()

for edge in edges:

    # Create A-id
    a_id = 'A' + str(edge['A'])

    # Add node with A-id to graph
    G.add_node(a_id)

    # Add value to node data
    G.nodes[a_id]['value'] = edge['A']

    # Create B-id
    b_id = 'B' + str(edge['B'])

    # Add node with B-id to graph
    G.add_node(b_id)

    # Add value to node data
    G.nodes[b_id]['value'] = edge['B']

    # Add A-id - B-id edge to graph
    G.add_edge(a_id, b_id)

如果你要绘制图形：

nx.draw(
    G,
    node_color='#FF0000',
    with_labels=True
)

你会得到这样的东西：