我想將分類變量的字符串數組轉換爲分類變量的整數數組。numpy將分類字符串數組轉換爲整數數組
Ex。
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
我意識到這可以用循環完成,但我想有一個更簡單的方法。謝謝。
你認真想添加的警告,這種方式只適用於長度爲1的字符串。 – smci 2013-07-23 12:23:39