如何解决使用欧几里得距离查找数组中最相似的索引
所以我有一个数组并给定了某一行,我想使用欧几里德距离输出最相似的行?我知道它在 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])
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。