python - 将同一行中的多个值连接到一个列表中

Question

我有一个 df，我想在其中进行多标签分类。向我建议的方法之一是计算概率向量。这是我的 DF 的示例，其中包含代表训练数据的内容。

    id  ABC DEF GHI
1   0   0   0   1
2   1   0   1   0
3   2   1   0   0
4   3   0   1   1
5   4   0   0   0
6   5   0   1   1
7   6   1   1   1
8   7   1   0   1
9   8   1   1   0

我想将 ABC、DEF、GHI 列连接成一个新列。我还必须对超过 3 列执行此操作，因此我想使用列列表或类似的内容相对干净地执行此操作:

col_list = ['ABC','DEF','GHI']

我正在寻找的结果是这样的:

    id  ABC DEF GHI Conc
1   0   0   0   1   [0,0,1]
2   1   0   1   0   [0,1,0]
3   2   1   0   0   [1,0,0]
4   3   0   1   1   [0,1,1]
5   4   0   0   0   [0,0,0]
6   5   0   1   1   [0,1,1]
7   6   1   1   1   [1,1,1]
8   7   1   0   1   [1,0,1]
9   8   1   1   0   [1,1,0]

Answer 1

尝试:

col_list = ['ABC','DEF','GHI']
df['agg_lst']=df.apply(lambda x: list(x[col] for col in col_list), axis=1)