在二維 numpy 中查找最近的一維數組

Question

我試圖從二維數組中找到一維數組的最佳匹配。

arr1 = np.array([[1, 1, 1], [2, 2, 2]])

arr2 = np.array([1.1, 1.1, 1.1])

我怎樣才能讓它返回最佳匹配，即行 0 最好作為行的索引？

Answer 1

假設“最接近”意味着最小的絕對差異，請嘗試以下操作：

idx = np.abs(arr1 - arr2).sum(axis=1).argmin()  # 0

絕對差異np.absolute ：

np.abs(arr1 - arr2)

[[0.1 0.1 0.1]
 [0.9 0.9 0.9]]

逐行總差和：

np.abs(arr1 - arr2).sum(axis=1)

[0.3 2.7]

閔指數argmin ：

np.abs(arr1 - arr2).sum(axis=1).argmin()

0