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

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

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

请您参考如下方法:

您可以使用 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})]