如何使用熊猫将多值行转换为列？

Question

我用什么函数来转换这个Dataframe：

id name genre

1 Fiml1 action, comedy
2 Fiml2 animation
3 Fiml3 comedy
4 Fiml4 action, animation
5 Fiml5 action
6 Fiml6 animation, comedy

收件人：

id name action animation comedy

1  Fiml1   1       0       1
2  Fiml2   0       1       0
3  Fiml3   0       0       1
4  Fiml4   1       1       0
5  Fiml5   1       0       0
6  Fiml6   0       1       1

此数据框将用于向量空间模型，答案或任何建议？

Answer 1

您可以使用pd.concat 和Series.str.dummies 和', ' 作为分隔符：

df = pd.concat([
    df[['id', 'name']], df['genre'].str.get_dummies(sep=', ')
], axis=1)

   id   name  action  animation  comedy
0   1  Fiml1       1          0       1
1   2  Fiml2       0          1       0
2   3  Fiml3       0          0       1
3   4  Fiml4       1          1       0
4   5  Fiml5       1          0       0
5   6  Fiml6       0          1       1