微信公众号搜"智元新知"关注
微信扫一扫可直接关注哦!

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

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

所以我有一个数组并给定了某一行,我想使用欧几里德距离输出最相似的行?我知道它在 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 举报,一经查实,本站将立刻删除。