将整数数组重新映射到 NumPy 数组中的范围

Question

假设我有一个 numpy 整数数组（可以是正数或负数，并且包含重复项），如下所示：

arr = array([-23, 1983, 2344, 323, -34, 323, -234432, 22343])

现在我想创建一个新数组arr_remapped，以便所有整数根据它们在数组中的大小得到一个新数字（从零开始），如下所示：

arr_remapped = array([2, 4, 5, 3, 1, 3, 0, 6])

arr_remapped 将包含[0, number_of_unique_numbers-1] 范围内的数字，并且可以包含重复项。

如何在几行代码中仅使用 NumPy（或者可能是 Scikit-learn）来做到这一点？

类似于this问题，但我只想在这里使用NumPy。

Answer 1

NumPy 中有一个函数：argsort。 https://docs.scipy.org/doc/numpy/reference/generated/numpy.argsort.html