如何从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})]