我有一個關于電影類型的數據。流派用“”分隔id genres0 drama romance1 drama2 comedy3 mystery thriller4 crime thriller...我想按類型劃分它們（大約 20 個）id drama romance comedy...0 1 1 01 1 0 02 0 0 13 0 0 04 0 0 0...我本來想買個假人，但我認為這沒有幫助。
1 回答
阿波羅的戰車
TA貢獻1862條經驗 獲得超6個贊
這是一個可能的解決方案（df是您的數據框）：
pd.merge(df[['id']], pd.get_dummies(df.genres.str.split().explode()),
left_on='id', right_index=True).groupby('id').sum()
這是一個例子：
>>> df = pd.DataFrame({'id': [0,1,2,3,4], 'genres': ['drama romance', 'drama', 'comedy', 'mystery thriller', 'crime thriller']})
>>> pd.merge(df[['id']], pd.get_dummies(df.genres.str.split().explode()), left_on='id', right_index=True).groupby('id').sum()
comedy crime drama mystery romance thriller
id
0 0 0 1 0 1 0
1 0 0 1 0 0 0
2 1 0 0 0 0 0
3 0 0 0 1 0 1
4 0 1 0 0 0 1
添加回答
舉報
0/150
提交
取消