如何將具有 n 個值的 DataFrame 列轉換為 n 個二進制列？

Question

我很確定有一個 Pandas 函數可以這樣做，但是當我查看文檔並用谷歌搜索時我找不到它。 也許你可以幫助我。 這是我想要做的：

import pandas as pd

data = {'col1': ['val1','val2','val3'],
        'col2': ['feat1','feat1','feat2']
       }
df = pd.DataFrame(data)
print (df)

>    col1   col2
> 0  val1  feat1
> 1  val2  feat1
> 2  val3  feat2

我想要這個形狀的數據框：

>    col1  feat1  feat2
> 0  val1      1      0
> 1  val2      1      0
> 2  val3      0      1

最佳P

Answer 1

print(pd.get_dummies(df.set_index('col1'), prefix='', prefix_sep=''))

印刷：

      feat1  feat2
col1              
val1      1      0
val2      1      0
val3      0      1

Answer 2

您也可以使用crosstab

pd.crosstab(df.col1, df.col2)