问题描述
所以我有一个数组并给定了某一行,我想使用欧几里德距离输出最相似的行?我知道它在 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])