python - 如何从 Networkx 中的边列表创建加权有向图

Question

我有一个边列表，它由两列组成，我想创建一个加权有向图，这样对于边列表中的每一行，权重为 1 的有向边从第一列中的节点到列中的节点二。如果同一行在边缘列表中出现不止一次，它应该在每次出现时将权重增加一。

我正在使用 Python Networkx 库，我该怎么做？

Answer 1

您可以使用 Counter计算重复边的数量以生成传递给 DiGraph 的权重:

import networkx as nx
from collections import Counter

EDGES = [
    ('A', 'B'),
    ('B', 'C'),
    ('A', 'C'),
    ('C', 'D'),
    ('A', 'B')
]

g = nx.DiGraph((x, y, {'weight': v}) for (x, y), v in Counter(EDGES).items())
print(*g.edges(data=True), sep='\n')

输出:

('A', 'B', {'weight': 2})
('A', 'C', {'weight': 1})
('C', 'D', {'weight': 1})
('B', 'C', {'weight': 1})

在上面的 Counter 中返回 (edge, count) 元组。请注意，传递给 Counter 的边必须是可哈希的。

>>> edges = list(Counter(EDGES).items())
>>> edges
[(('A', 'B'), 2), (('B', 'C'), 1), (('C', 'D'), 1), (('A', 'C'), 1)]

然后使用生成器表达式以 DiGraph 期望的格式生成边:

>>> params = list((x, y, {'weight':v}) for (x,y), v in edges)
>>> params
[('A', 'B', {'weight': 2}), ('B', 'C', {'weight': 1}), ('C', 'D', {'weight': 1}), ('A', 'C', {'weight': 1})]