假设我有一个国家的数据框,如下所示:
cc | temp
US | 37.0
CA | 12.0
US | 35.0
AU | 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中使用'factors'更容易做到,所以我希望pandas也有类似的东西。
6条答案
按热度按时间raogr8fs1#
首先,更改列的类型:
现在,数据看起来相似,但分类存储。要捕获类别代码:
现在您有:
如果你不想修改你的DataFrame,而只是获取代码:
或者使用分类列作为索引:
uklbhaso2#
如果只想将序列转换为整数标识符,可以使用
pd.factorize
。请注意,与
pd.Categorical
不同,此解决方案不会按字母顺序排序。因此第一个国家将被分配0
。如果您希望从1
开始,您可以添加一个常量:如果希望按字母顺序排序,请指定
sort=True
:k4aesqcs3#
如果你使用的是
sklearn
库,你可以使用LabelEncoder
。像pd.Categorical
一样,输入字符串在编码前按字母顺序排序。ahy6op9u4#
单行代码:
如果你有一个
list_of_columns
,这也是有效的:此外,如果你想保留你的
NaN
值,你可以应用一个替换:de90aj5v5#
试试这个,根据频率转换为数字(高频-高数字):
kxkpmulp6#
将任何列更改为Numbers。它不会创建新列,而只是将值替换为数值数据。