熊貓：將類別轉換為數字假設我有一個數據框，其國家/地區如下：cc | tempUS | 37.0CA | 12.0US | 35.0AU | 20.0我知道有一個pd.get_dummies函數可以將這些國家/地區轉換為“單熱編碼”。但是，我希望將它們轉換為索引，以便我得到它cc_index = [1,2,1,3]。我假設有一種比使用get_dummies和numpy where子句更快的方法，如下所示：[np.where(x) for x in df.cc.get_dummies().values]使用'因子'在R中這樣做有點容易，所以我希望大熊貓有類似的東西。
3 回答
浮云間
TA貢獻1829條經驗 獲得超4個贊
如果您只希望將系列轉換為整數標識符，則可以使用pd.factorize。
請注意，此解決方案pd.Categorical不會按字母順序排序。所以第一個國家將被分配0。如果您想從頭開始1，可以添加常量：
df['code'] = pd.factorize(df['cc'])[0] + 1
print(df)
cc temp code
0 US 37.0 1
1 CA 12.0 2
2 US 35.0 1
3 AU 20.0 3
如果您希望按字母順序排序，請指定sort=True：
df['code'] = pd.factorize(df['cc'], sort=True)[0] + 1
添加回答
舉報
0/150
提交
取消