熊貓：DataFrame的嵌套迭代

Question

我有一個數據框，其中包含兩列：'rep_id'和'值'。第二列是數字列表。

我想計算'rep_id'的所有可能的對，然後找到對這些對通用的值。

最終的數據框將如下所示：'rep_id1'，'rep_id2'，'values'其中'values'表示'rep_id1'和'rep_id2'列表之間的公共元素列表。

做到這一點的一種方法是創建一個嵌套循環，並通過rep_ids對，但我似乎無法找到一種方法來做到這一點。

Answer 1

http://en.wikipedia.org/wiki/Cartesian_product

import pandas as pd 

df = pd.DataFrame(zip(['A','B','C','D','E','A','B','C','A'],[1,2,3,4,5,2,3,5,5]),columns=['rep_id','val']) 

df = df.sort('val') 

cartesian_product = pd.merge(df, df,on='val') 
cartesian_product = cartesian_product[cartesian_product['rep_id_x'] != cartesian_product['rep_id_y']] 
cartesian_product[['val','rep_id_x','rep_id_y']] 

DF：

rep_id val 
0  A 1 
1  B 2 
5  A 2 
2  C 3 
6  B 3 
3  D 4 
4  E 5 
7  C 5 
8  A 5 

cartesian_product：

val rep_id_x rep_id_y 
2  2  B  A 
3  2  A  B 
6  3  C  B 
7  3  B  C 
11 5  E  C 
12 5  E  A 
13 5  C  E 
15 5  C  A 
16 5  A  E 
17 5  A  C