python - 字符串操作，然后对其应用虚拟变量

Question

我正在开发一个小项目来构建推荐系统。我有电影的类别列，我想对其应用虚拟变量。

#Here is my data
movie_data.head()

Customer  MovieID  Movie_title       Movie_Category  Movie_cost($)

C1          M1     Gladiator         Drama,Action        3

C2          M2     The Accountant    Drama,Crime         2


#Applying the dummy variables
df_dummies = pd.get_dummies(movie_data['Movie_Category'])
df_dummies.head()

#Here's the output:
Drama,Action   Drama,Crime

1                   0

0                   1

#the desire output:
Drama   Action  Crime

1          1      0

1          0      1

Answer 1

使用Series.str.get_dummies :

df_dummies = movie_data['Movie_Category'].str.get_dummies(',')