在另一個數組中找到一個 numpy 數組的行列的最有效方法是什么？

Question

假設我有兩個數組：

a = np.array([2, 5, 4, 3, 1])
b = np.array([4.5, 1.5, 3.5])

如果它在b中，我想找到a的每個元素的等級。 所以這將是所需的輸出：

[1, 3, 2, 1, 0]

以下代碼在技術上適用於小型數組，但如果a和b的大小為 10,000+，則速度非常慢：

ranks = [rankdata(b + [i])[-1] - 1 for i in a]

實現這一結果的最有效方法是什么？

Answer 1

使用searchsorted

ind = np.argsort(y)
np.searchsorted(y, x, sorter=ind)
# array([1, 3, 2, 1, 0], dtype=int64)

如果y有重復項，您可能需要修改side參數。