Как Горячее Кодировать с Пандами, не комбинируя уровни строк

Question

Я создал действительно большой фрейм данных в pandas, подобный следующему:

    0   1
user        
0   product4    product0
1   product3    product1

Я хочу использовать что-то вроде pd.get_dummies() таким образом, чтобы конечный df был бы похож на:

    product0    product1    product2    product3  product4
user                
0   1   0   0   0   1
1   0   1   0   1   0

вместо получения следующего от pd.get_dummies():

    0_product3  0_product4  1_product0  1_product1
user                
0   0   1   1   0
1   1   0   0   1

Итак, я не хочу, чтобы строки объединялись в двоичные столбцы. Большое спасибо!

Answer 1

df = pd.get_dummies(df, prefix='', prefix_sep='') # remove prefix from dummy column names and underscore
df = df.sort_index(axis=1) # order data by column names

Answer 2

Используйте reindex с get_dummies

In [539]: dff = pd.get_dummies(df, prefix='', prefix_sep='')

In [540]: s = dff.columns.str[-1].astype(int)

In [541]: cols = 'product' + pd.RangeIndex(s.min(), s.max()+1).astype(str)

In [542]: dff.reindex(columns=cols, fill_value=0)
Out[542]:
      product0  product1  product2  product3  product4
user
0            1         0         0         0         1
1            0         1         0         1         0