我有一個 id 變數為 Toy 的資料框，不同的配色方案 Toy 進來 -
input_data = pd.DataFrame({'Toy': ['Toy1', 'Toy2', 'Toy3','Toy4'],
'Color1': ['Red', 'Orange', '', 'Orange'],
'Color2': ['Red', '', 'Blue', 'Red']})
我想對 Color1 和 Color2 變數進行一次熱編碼，但只有一個轉換后的變數（顏色名稱沒有任何前綴）
output_data = pd.DataFrame({'Toy': ['Toy1', 'Toy2', 'Toy3', 'Toy4'],
'Red': [1, 0, 0, 1],
'Blue': [0, 0, 1, 0],
'Orange': [0, 1, 0, 1]})
這似乎是一個快速簡便的輸出，但我無法找到直接的方法。任何線索都非常感謝。
uj5u.com熱心網友回復：
通過以下Series.str.get_dummies方式與連接值一起使用|：
df = input_data.set_index('Toy').agg('|'.join, 1).str.get_dummies().reset_index()
print (df)
Toy Blue Orange Red
0 Toy1 0 0 1
1 Toy2 0 1 0
2 Toy3 1 0 0
3 Toy4 0 1 1
uj5u.com熱心網友回復：
我認為沒有直接的選擇。您可以重塑和使用crosstab：
d = input_data.replace('', float('nan')).melt(id_vars='Toy')
out = (pd.crosstab(d['Toy'],d['value'])
.clip(upper=1)
.reset_index().rename_axis(index=None, columns=None)
)
輸出：
Toy Blue Orange Red
0 Toy1 0 0 1
1 Toy2 0 1 0
2 Toy3 1 0 0
3 Toy4 0 1 1
轉載請註明出處，本文鏈接：https://www.uj5u.com/ruanti/424776.html