将NumPy数组矢量化重新标记为连续数字并取回

将NumPy数组矢量化重新标记为连续数字并取回

我们可以使用可选参数

return_inverse

with

np.unique

来获取那些唯一的顺序ID
/标记，如下所示-

unq_arr, unq_tags = np.unique(old_classes,return_inverse=1)

索引为

unq_arr

with

unq_tags

以取回-

old_classes_retrieved = unq_arr[unq_tags]

样品运行-

In [69]: old_classes = np.array([0,1,2,6,6,2,6,1,1,0])In [70]: unq_arr, unq_tags = np.unique(old_classes,return_inverse=1)In [71]: unq_arrOut[71]: array([0, 1, 2, 6])In [72]: unq_tagsOut[72]: array([0, 1, 2, 3, 3, 2, 3, 1, 1, 0])In [73]: old_classes_retrieved = unq_arr[unq_tags]In [74]: old_classes_retrievedOut[74]: array([0, 1, 2, 6, 6, 2, 6, 1, 1, 0])