熊猫-还原许多热编码（虚拟变量）

Question

假设我有以下pandas.DataFrame：

df = pd.DataFrame({'id': [1, 2, 3], 'val': [5, 5, 10],
                   'trig_aaa': [1, 0, 1], 'trig_bbb': [0, 1, 1], 'trig_ccc': [0, 0, 1]})
print(df)

   id  val  trig_aaa  trig_bbb  trig_ccc
0   1    5         1         0         0
1   2    5         0         1         0
2   3   10         1         1         1

我想将其转换为以下df：

   id  val             trig
0   1    5            [aaa]
1   2    5            [bbb]
2   3   10  [aaa, bbb, ccc]

Pandas / Python / Numpy中是否有精美的（希望是预先构建的功能）？

编辑1：

在查看jpps的注释之后，对DataFrame的更好处理如下所示：

   id  val trig
0   1    5  aaa
1   2    5  bbb
2   3   10  aaa
3   3   10  bbb
4   3   10  ccc

Answer 1

您可以使用pd.melt：

# rename columns and melt dataframe
df.columns = [i if '_' not in i else i.split('_')[1] for i in df]
res = pd.melt(df, id_vars=['id', 'val'], var_name='trig')

# filter for 1 values and sort
res = res[res['value'].eq(1)].sort_values('id').iloc[:, :-1].reset_index(drop=True)

print(res)

   id  val trig
0   1    5  aaa
1   2    5  bbb
2   3   10  aaa
3   3   10  bbb
4   3   10  ccc