Преобразование кадра панды данных в кадре данных панд с другим стилем

Question

У меня есть кадр данных, содержащий идентификаторы животных и типов, которым они принадлежат, как указано ниже

ID Class 
1 1 
2 1 
3 0 
4 4 
5 3 
6 2 
7 1 
8 0 

Я хочу превратить его в новый стиль с классов в строке заголовка следующим образом.

ID 0 1 2 3 4 
1  1   
2  1   
3 1    
4     1 
5    1 
6   1  
7  1   
8 1    

Можете ли вы помочь мне сделать это с питоном

Answer 1

См get_dummies():

>>> print df 

    ID Class 
0 1  1 
1 2  1 
2 3  0 
3 4  4 
4 5  3 
5 6  2 
6 7  1 
7 8  0 

>>> df2 = pd.get_dummies(df, columns=['Class']) 
>>> print df2 

    ID Class_0 Class_1 Class_2 Class_3 Class_4 
0 1  0  1  0  0  0 
1 2  0  1  0  0  0 
2 3  1  0  0  0  0 
3 4  0  0  0  0  1 
4 5  0  0  0  1  0 
5 6  0  0  1  0  0 
6 7  0  1  0  0  0 
7 8  1  0  0  0  0 

И если вы хотите, чтобы избавиться от «Class_» в заголовке столбца, установите как prefix и prefix_sep до пустой строки:

df2 = pd.get_dummies(df, columns=['Class'], prefix='', prefix_sep='')