python - Pandas - 恢复许多热编码(虚拟变量)

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