use*_*980 5 python graph-theory networkx
假设我在NetworkX中创建了一个DiGraph:
import networkx as nx
G = nx.DiGraph()
n = ["A","B","C","D","E","F","H","I","J","K","L","X","Y","Z"]
e = [("A","Z"),("Z","B"),("B","Y"),("Y","C"),("C","G"),("G","H"),("G","I"),("I","J"),("K","J"),("J","L"),("F","E"),("E","D"),("D","X"),("X","C")]
G.add_nodes_from(n)
G.add_edges_from(e)
Run Code Online (Sandbox Code Playgroud)
如何删除所有具有in-degree和out-degree等于1的节点,以便我的图形看起来像这样?:
import networkx as nx
G = nx.DiGraph()
n = ["A","C","F","G","H","J","K","L"]
e = [("A","C"),("C","G"),("G","H"),("G","J"),("K","J"),("J","L")
G.add_nodes_from(n)
G.add_edges_from(e)
Run Code Online (Sandbox Code Playgroud)
我们的想法是删除"流通"节点并保持连接.
以下代码执行您想要的操作,但我在最终结果中看不到您将从哪里获得边("A","C")和("G","J").
import networkx as nx
def remove_edges(g, in_degree=1, out_degree=1):
g2=g.copy()
d_in=g2.in_degree(g2)
d_out=g2.out_degree(g2)
print(d_in)
print(d_out)
for n in g2.nodes():
if d_in[n]==in_degree and d_out[n] == out_degree:
g2.remove_node(n)
return g2
G_trimmed = remove_edges(G)
G_trimmed.edges()
#outputs [('C', 'G'), ('G', 'H'), ('K', 'J'), ('J', 'L')]
G_trimmed.nodes()
#outputs ['A', 'C', 'G', 'F', 'H', 'K', 'J', 'L']
Run Code Online (Sandbox Code Playgroud)
| 归档时间: |
|
| 查看次数: |
2565 次 |
| 最近记录: |