我有一个看起来像..的DataFrame
ID c1 c2 cX
r1 2 3 ..
r2 8 9 ..
rY ..
我想生成一个新的DataFrame,其中包含所有可能的(两部分)行组合,同时连接两个组合行的列(以便新DF将具有两倍的列)。结果应如下所示:
ID c1_r1 c1_r2 c2_r1 c2_r2 cX_rA
r1_r2 2 8 3 9 ..
r1_r3 .. .. .. ..
rA_rB ..
ID名称并不重要(它甚至可能是MultiIndex),也不是重要列的顺序。
怎么解决这个问题?
考虑一下df
c1 c2
ID
r1 2 3
r2 8 9
r3 0 7
我会这样做的
from itertools import combinations
a, b = map(list, zip(*combinations(df.index, 2)))
print(a, b, sep='\n')
['r1', 'r1', 'r2']
['r2', 'r3', 'r3']
然后使用pd.concat
d = pd.concat(
[df.loc[a].reset_index(), df.loc[b].reset_index()],
keys=['a', 'b'], axis=1
)
d
a b
ID c1 c2 ID c1 c2
0 r1 2 3 r2 8 9
1 r1 2 3 r3 0 7
2 r2 8 9 r3 0 7
最后,绑定松散的末端
d.set_index([('a', 'ID'), ('b', 'ID')]).rename_axis(['a', 'b'])
a b
c1 c2 c1 c2
a b
r1 r2 2 3 8 9
r3 2 3 0 7
r2 r3 8 9 0 7