將雙向圖邊轉換為 NetworkX 中的無向邊

Question

我正在使用NetworkX執行以下操作：我有一個有向圖g ，我想將其轉換為一個無向圖h ，使得h具有與g相同的節點，並且e是h中的一條邊，當且僅當e在g中是雙向的。 例如，我想轉換：

import networkx as nx
g = nx.DiGraph([("A", "B"), ("B", "A"), ("A", "C"), ("C", "A"), ("B", "C"), ("C", "B"), ("A", "D"), ("C", "D"), ("B", "E"), ("C", "E")])

進入

h = nx.Graph([("A", "B"), ("A", "C"), ("B", "C")])
h.add_nodes_from(["D", "E"])

我應該在 NetworkX 中寫什么f(g) = h ？ 我認為它是圖形視圖和過濾器的某種組合，但我是 NetworkX 的新手，所以我不確定到底是什么。

Answer 1

您可以通過迭代有向圖的邊緣並檢查反向邊緣是否存在條件if edge[::-1] in g.edges(): 。 如果反向邊緣存在，只需將其添加到您的圖形中。 請參閱下面的代碼：

import networkx as nx
import matplotlib.pyplot as plt

#Creating directed graph
g = nx.DiGraph([("A", "B"), ("B", "A"), ("A", "C"), ("C", "A"), ("B", "C"), ("C", "B"), ("A", "D"), ("C", "D"), ("B", "E"), ("C", "E")])

#Creating undirected graph
h=nx.Graph()
h.add_nodes_from(g)

for edge in g.edges():  
  if edge[::-1] in g.edges(): #check if reverse edge exist in graph
    h.add_edge(edge[0],edge[1])

#plot both graphs
fig=plt.figure(figsize=(15,6))
plt.subplot(121)
plt.title('Directed graph')
pos1=nx.circular_layout(g)
nx.draw(g,pos=pos1,with_labels=True, node_color='tab:orange')

plt.subplot(122)
plt.title('Undirected graph')
pos2=nx.circular_layout(h)
nx.draw(h,pos=pos2,with_labels=True, node_color='tab:green')

output 給出：