将NumPy数组的矢量化重新标记为连续数并重新检索

我们可以使用可选的参数return_inverse用np.unique得到这些独特的顺序ID /标签,像这样-

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

索引unq_arr与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_arr
Out[71]: array([0, 1, 2, 6])

In [72]: unq_tags
Out[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_retrieved
Out[74]: array([0, 1, 2, 6, 6, 2, 6, 1, 1, 0])