您好, 欢迎来到 !    登录 | 注册 | | 设为首页 | 收藏本站

计算numpy中2个点列表的距离

计算numpy中2个点列表的距离

使用直接的numpy广播,您可以执行以下操作:

dist = np.sqrt(((a[:, None] - b[:, :, None]) ** 2).sum(0))

另外,scipy一个例程可以稍微提高效率(特别是对于大型矩阵)

from scipy.spatial.distance import cdist
dist = cdist(a, b)

我将避免依赖于分解矩阵乘积(形式为A ^ 2 + B ^ 2-2AB)的解决方案,因为由于浮点舍入误差,它们在数值上可能不稳定。

其他 2022/1/1 18:43:23 有497人围观

撰写回答


你尚未登录,登录后可以

和开发者交流问题的细节

关注并接收问题和回答的更新提醒

参与内容的编辑和改进,让解决方法与时俱进

请先登录

推荐问题


联系我
置顶