我有一个数据框,其中有两列,我们称其为A和B,我想找到所有对称的配对,例如:
pd.DataFrame({'A':[1, 2 , 3], 'B':[2, 1, 3]})
我希望能够返回所有这样的配对,就像这样:我将得到(实际上我只需要其中一个,只要有(2,1)或(1,2)即可)。
(1,2) and (2,1)
我首先尝试了一种算法,虽然能够工作但在我的数据框长度为26325时速度非常慢,在10分钟后仍未完成。
listTuples = list()
for index, row in test.iterrows():
listTuples.append((row["A"], row["B"])) # convert to a list of tuple
answer = [(x, y) for (x, y) in listTuples if (y, x) in listTuples]