![](/img/trans.png)
[英]Python Reading from a file to create a weighted directed graph using networkx
[英]how to created a weighted directed graph from edge list in 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})]
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.