Numpy:将标签转换为索引

Question

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

假设我有一个字符串数组，比如['ABC', 'DEF', 'GHI', 'DEF', 'ABC']等等，我希望它变成一个像[0,1,2,1,0]这样的整数数组。有可能使用numpy吗？我知道Pandas有一个Series类可以做到这一点，这是由{a1}提供的。numpy也有类似的情况吗？在

编辑： np.unique()返回所有元素的唯一值。我要做的是将Iris dataset中的标签转换为索引，例如0代表Iris-setosa，1代表Iris-versicolor，2代表Iris-virginica。有没有一种方法可以使用numpy？在

Answer 1

将^{}与参数return_inverse=True一起使用，但与处理NaNs-checkfactorizing values有区别：

L = ['ABC', 'DEF', 'GHI', 'DEF', 'ABC']

print (np.unique(L, return_inverse=True)[1])
[0 1 2 1 0]

pandas^{}使用列表或数组也很好：

^{pr2}$