将int编号分配给数组

Question

我想将int分配给数组。我所拥有的;

label = ['rest', 'rest', 'ball', 'cat', 'rest']

我想得到这样的东西，int顺序并不重要;

labeled = [1, 1, 2, 3, 1]

我的工作方式，我找到了NP. Unique（标签）并制作dict：

labelDict = dict(zip(label, np.arange(1,4))

但是，我陷入了下一步，这是将此整数分配给该数组标签。

Answer 1

您可以创建一个映射字典，类似于您正在做的事情：

mapping = {i:idx for idx, i in enumerate(set(label))}

然后，使用映射重塑标签：

output = [mapping[i] for i in label]

不需要Numpy，可以使用Base Pyhton轻松完成任务。