python - 使用 Pandas 进行反向二进制编码

Question

我有一个 Pandas Dataframe，希望反转三列的二进制编码(即 get_dummies())。编码是从左到右:

    a   b   c
0   0   1   1
1   0   0   1
2   1   1   1
3   1   0   0

将导致一个新的类别列 C 取值 0-7:

    C
1   6   
2   4   
3   7
4   1

我不确定为什么这一行在 axis=1 附近给我一个语法错误:

df['C'] = df.apply(lambda x: (x['a']==1 ? 1:0)+(x['b']==1 ? 2:0)+(x['c']==1 ? 4:0), axis=1)

Answer 1

如果性能很重要，请使用 numpy - 首先将 DataFrame 转换为 numpy 数组，然后使用 bitwise shift :

a = df.values
#pandas 0.24+
#a = df.to_numpy()
df['C'] = a.dot(1 << np.arange(a.shape[-1]))
print (df)
   a  b  c  C
0  0  1  1  6
1  0  0  1  4
2  1  1  1  7
3  1  0  0  1