替换数组中的元素（python）

Question

我有一个数组：

y=['a','s','d','a','f','d','g']

我想用整数替换此数组的所有元素。 我认为一种简单的解决方案是执行以下操作：

c = np.unique(y)

，然后将y中的所有元素替换为c中的索引。 我可以这样进行元素比较：

for n, i in enumerate(c):
    for m, j in enumerate(y):
        if i == j:
            y[m] = n

输出：

 y=[0, 4, 1, 0, 2, 1, 3]

但是是否有一个numpy函数以紧凑的方式执行此操作？

Answer 1

y = [c.tolist().index(i) for i in y]