python - 基于python中的不同二进制列创建分类列

Question

我有一个看起来像这样的数据集:

df = pd.DataFrame(data= [[0,0,1],[1,0,0],[0,1,0]], columns = ['A','B','C'])

    A   B   C
0   0   0   1
1   1   0   0
2   0   1   0

我想创建一个新列，其中每一行都显示前一列的值，其中有一个 1:

    A   B   C value
0   0   0   1   C
1   1   0   0   A
2   0   1   0   B

Answer 1

使用点:

df['value'] = df.values.dot(df.columns)

输出:

   A  B  C value
0  0  0  1     C
1  1  0  0     A
2  0  1  0     B