我们可以使用可选参数
return_inversewith
np.unique来获取那些唯一的顺序ID
/标记，如下所示-
unq_arr, unq_tags = np.unique(old_classes,return_inverse=1)
索引为
unq_arrwith
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])
