pandas 数据框行的所有可能组合

【问题标题】：All possible combinations of pandas data frame rowspandas 数据框行的所有可能组合
【发布时间】：2018-08-08 12:22:19
【问题描述】：

我有一个有 4 行的 pandas 数据框：

df:

col1    col2    col3    col4
A1      A2      A3      A4
B1      B2      B3      B4 
C1      C2      C3      C4
D1      D2      D3      D4

如何找到选择此数据框的两行的所有可能组合。在这种情况下，我可以在 4C2 = 6 种可能的方式中从 4 行中选择 2 行

df1:

col1    col2    col3    col4
A1      A2      A3      A4
B1      B2      B3      B4

df2:

col1    col2    col3    col4
A1      A2      A3      A4
C1      C2      C3      C4

df3:

col1    col2    col3    col4
A1      A2      A3      A4
D1      D2      D3      D4

等等……

【问题讨论】：

标签： python pandas combinations

【解决方案1】：

首先，您需要使用itertools 查找所有组合，然后使用combinations 的输出作为数据框的索引。您将获得给定行数的所有可能数据帧。

from itertools import combinations
for index in list(combinations(df.index,2)):
    print(df.loc[index,:])
    print('\n')

输出将是：

  col1 col2 col3 col4
0   A1   A2   A3   A4
1   B1   B2   B3   B4


  col1 col2 col3 col4
0   A1   A2   A3   A4
2   C1   C2   C3   C4


  col1 col2 col3 col4
0   A1   A2   A3   A4
3   D1   D2   D3   D4


  col1 col2 col3 col4
1   B1   B2   B3   B4
2   C1   C2   C3   C4


  col1 col2 col3 col4
1   B1   B2   B3   B4
3   D1   D2   D3   D4


  col1 col2 col3 col4
2   C1   C2   C3   C4
3   D1   D2   D3   D4

【讨论】：