如何在Networkx中从边列表创建加权有向图

Question

我有一个边列表，它由两列组成，我想创建一个带权有向图，使得对于边列表中的每一行，从第一列中的节点到第二列中的节点都有一个权重为1的有向边。如果同一行在边列表中出现多次，则应每次增加1个权重。

我使用Python Networkx库，如何实现这个功能？

- Ramin Zahedi

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})]

- niemmi