我试图找出图中的哪些边是双向的。每一行都是一条边。对于每个起始节点 A,如果它们以节点 A 作为结束点,我将搜索每个相应的结束节点 B:
for ending_point_B in nodeA:
nodeA in ending_points_of_B
现在忽略 df['S'] 中的重复条目。如何优化此搜索?我怀疑类似于 groupby 的东西。这种方式对于我的真实图表来说需要太多时间。
谢谢
from pandas import *
def missing_node(node):
set1 = set(df[df.E == node].S.values)
set2 = set(df.E[df.S == node].values)
return list(set1.difference(set2))
x = [1,1,2,2,3]
y = [2,3,1,3,1]
df = DataFrame([x,y]).T
df.columns = ['S','E'] #Start & End
df['Missing'] = df.S.apply(missing_node)
df:
S E Missing
0 1 2 []
1 1 3 []
2 2 1 []
3 2 3 []
4 3 1 [2]