Question

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

我使用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})]

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

1 个答案: