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