使用欧几里得距离查找数组中最相似的索引

问题描述

所以我有一个数组并给定了某一行,我想使用欧几里德距离输出最相似的行?我知道它在 scipy.spatial.distance 中,但我不知道如何实现它!

例如:

S = np.array([[1,10,2,5,1],[10,3,1,3],[2,9,8,5],[9,4,4],[8,7,[7,8]])

row = [7,8]

输出最相似的行是 5

TIA!

解决方法

通过使用带有 for 参数的 numpy 的 linalg.norm 函数,您可以非常快速地完成(没有隐藏在理解列表中的缓慢的 axis=1 循环):

row_id = np.argmin(np.linalg.norm(S-row,axis=1))

print(S[row_id]) # Returns array([7,2,9,8])