python - 如何获取数据透视表中的连接值？

Question

我目前拥有这样的数据集-

    customerID  store_code  mode
    BBID_204100102  2655    a
    BBID_204100102  2906    b
    BBID_204100102  2906    d 
    BBID_204100150  4986    c      
    BBID_204100150  4986    a
    BBID_204100277  4986    d                                                 
    BBID_204100310  4986    d

我想要类似的东西

    customerID  store_code  a   b  c  d
0   BBID_204100102  2655    1   0   0  0
1   BBID_204100102  2906    0   1   0  0
2   BBID_204100150  4986    1   0   1  0 
3   BBID_204100277  4986    0   0   0  1
4   BBID_204100310  4986    0   0   0  1

首先对客户 ud 和学生 ID 进行数据透视，然后以上述方式对模式进行编码。

Answer 1

使用get_dummies与 set_index和 max 两个级别:

df = (pd.get_dummies(df.set_index(['customerID','store_code']), prefix='', prefix_sep='')
        .max(level=[0,1])
        .reset_index())
print (df)
       customerID  store_code  a  b  c  d
0  BBID_204100102        2655  1  0  0  0
1  BBID_204100102        2906  0  1  0  1
2  BBID_204100150        4986  1  0  1  0
3  BBID_204100277        4986  0  0  0  1
4  BBID_204100310        4986  0  0  0  1