Pandasでコードを書いていると、ワンホットエンコーディングしたい場合が時々あります。
例えば、以下のようなデータを扱うときにGenre列をワンホットにしたい。って時ありますよね。
pandasのget_dummiesを使うと、
|
1 |
pd.get_dummies(df_movie['Genre']) |
「あら、そうじゃなくって、|で区切っているジャンルごとだよ。」
って時には以下のようにすると良い。
|
1 |
df_movie['Genre'].str.get_dummies(sep='|') |
元のデータフレームに結合する時には、concatを使おう。
|
1 |
df_movie = pd.concat([df_movie, df_movie['Genre'].str.get_dummies(sep='|')], axis=1) |
元のカラムが必要ない場合は、
|
1 |
df_movie.drop('Genre', axis=1) |
で削除できるよ。
おわり。
