在圖表中只保留列表中的那些邊

Question

我有這個名為edgeKeep的邊（元組）列表，我想在圖中保留它，並刪除所有其他邊。 在我的情況下（因為edgeKeep的構建方式），不能保證edgeKeep邊緣中的節點名稱遵循G.edges中節點名稱的相同順序（邊緣可以是('u','v')一個並在另一個中顯示為('v','u') ）。 所以我必須這樣做：

G.remove_edges_from(e for e in G.edges if e not in edgesKeep and tuple(reversed(e)) not in edgesKeep)

有沒有辦法讓它更簡單？ 我的意思是，以某種方式擺脫tuple(reversed(e)) not in edgesKeep ？

Answer 1

您可以對邊中的元組進行edgesKeep並執行

G.remove_edges_from(e for e in G.edges if sorted(e) not in edgesKeep)