我在 NetworkX 中有一个有向图。边的权重从 0 到 1,表示它们发生的概率。网络连接性非常高,所以我想为每个节点修剪边缘,只剩下概率最高的节点。
我不确定如何迭代每个节点并只保留图中权重最高的节点in_edges
。是否有允许我们这样做的 networkx 功能?
这是我想做的一个例子。
Nodes:
A, B, C, D
Edges:
A->B, weight=1.0
A->C, weight=1.0
A->D, weight=0.5
B->C, weight=0.9
B->D, weight=0.8
C->D, weight=0.9
Final Result Wanted:
A->B, weight=1.0
A->C, weight=1.0
C->D, weight=0.9
如果一个节点有两条边,并且它们的权重都最高,我想保留它们。