相关疑难解决方法(0)

什么是最接近相当于R因素变量在Python的大熊猫？

我正在尝试将分类变量的字符串数组转换为分类变量的整数数组.

防爆.

import numpy as np
a = np.array( ['a', 'b', 'c', 'a', 'b', 'c'])
print a.dtype
>>> |S1

b = np.unique(a)
print b
>>>  ['a' 'b' 'c']

c = a.desired_function(b)
print c, c.dtype
>>> [1,2,3,1,2,3] int32

我意识到这可以通过循环完成,但我想有一种更简单的方法.谢谢.

是否可以使用将字符串向量转换为索引向量numpy？

假设我有一个字符串数组，例如['ABC', 'DEF', 'GHI', 'DEF', 'ABC']etc。我希望将其更改为一个整数数组，例如[0,1,2,1,0]。是否可以使用numpy？我知道Pandas有一个Series班级可以做到这一点，这要归功于这个答案。还有类似的东西numpy吗？

编辑： np.unique()返回所有元素的唯一值。我正在试图做的是转换标签的虹膜数据集，以指数，如0 Iris-setosa为，1 Iris-versicolor和2 Iris-virginica分别。有没有办法做到这一点numpy？