python - 如何计算所有可能的列组合的总和

Question

我有以下 df:

df = pd.DataFrame({'a': [1,2,3,4,2], 'b': [3,4,1,0,4], 'c':[1,2,3,1,0], 'd':[3,2,4,1,4]})

我想生成这 4 列的总计组合，等于 4 x 3 x 2 = 24 总组合减去重复项。我想要相同 df 中的结果。

我想要这样的东西(显示部分结果):

a_b 的组合与 b_a 相同，因此我不想要这样的计算，因为它是重复的。

有没有办法计算所有的组合并排除重复的总数？

Answer 1

import itertools as it

orig_cols = df.columns
for r in range(2, df.shape[1] + 1):
    for cols in it.combinations(orig_cols, r):
        df["_".join(cols)] = df.loc[:, cols].sum(axis=1)

需要一些循环，但不是在数据帧本身上，而是在组合上。我们得到列名的第 2、3、...、N-1 种组合，其中 N 是列数。然后形成新的 _-joined 列作为总和。

In [11]: df
Out[11]:
   a  b  c  d  a_b  a_c  a_d  b_c  b_d  c_d  a_b_c  a_b_d  a_c_d  b_c_d  a_b_c_d
0  1  3  1  3    4    2    4    4    6    4      5      7      5      7        8
1  2  4  2  2    6    4    4    6    6    4      8      8      6      8       10
2  3  1  3  4    4    6    7    4    5    7      7      8     10      8       11
3  4  0  1  1    4    5    5    1    1    2      5      5      6      2        6
4  2  4  0  4    6    2    6    4    8    4      6     10      6      8       10