所有可能的组合作为基于主列的数据框的新列

Question

我有一个这样的数据框

Dt              A           B           C          D
11-apr          1           1           1          1
10-apr          2           3           1          2

如何获得如下所示的新框架：

我想要ABCD所有可能的组合

可能的组合：A+B、A+C、A+D、B+c、B+d、c+D、A+B+C、B+C+D、A+B+D、 A+C+D,A+B+C+d

我能够获得所有组合，但我得到了文件 "pandas/_libs/src/inference.pyx"，第 1472 行，在 pandas._libs.lib.map_infer TypeError：序列项 2：预期字符串，找到浮点数当我尝试创建这些列时。

from itertools import chain, combinations
ss_list1=[]
ss_list2=[]
for subset in all_subsets(df):
    ss_list=list(subset)
    # print ss_list
    ss_list_t=tuple(ss_list)
    # print ss_list_t
    ss_list1.insert(1,ss_list_t)
for c in ss_list1:
    if len(c)>1:
        # print c
        ss_list2.insert(1, c)
print ss_list2

df = pd.concat([df[c[1]].add(df[c[0]]) for c in ss_list2], axis=1, keys=ss_list2)
df.columns = df.columns.map(','.join)

文件 "pandas/_libs/src/inference.pyx"，第 1472 行，在 pandas._libs.lib.map_infer 中 TypeError：序列项 2：预期字符串，找到浮点数

Answer 1

使用：

#create index from Dt column if necessary
df = df.set_index('Dt')

#
from itertools import chain, combinations
def all_subsets(ss):
    return chain(*map(lambda x: combinations(ss, x), range(2, len(ss)+1)))

#get all combination from 2 to N
tups = list(all_subsets(df.columns))
#for each combination sum values
df1 = pd.concat([df.loc[:,c].sum(axis=1) for c in tups], axis=1)
#set new columns by join list of tuples tups
df1.columns = ['+'.join(x) for x in tups]
print (df1)
        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
Dt                                                                       
11-apr    2    2    2    2    2    2      3      3      3      3        4
10-apr    5    3    4    4    5    3      6      7      5      6        8

所有可能的组合作为基于主列的数据框的新列

All possible combinations as new columns of a dataframe based on primary column

python

combinations

probability

pandas