NumPy广播:计算两个数组之间的平方差之和

NumPy广播:计算两个数组之间的平方差之和,第1张

NumPy广播:计算两个数组之间的平方差之和

您可以

np.einsum
在计算出的差异后使用
broadcastedway
,如下所示:

ab = a[:,None,:] - bout = np.einsum('ijk,ijk->ij',ab,ab)

或使用

scipy'scdist
其可选的metric参数集,
'sqeuclidean'
以根据问题的需要给我们平方的欧几里得距离,如下所示-

from scipy.spatial.distance import cdistout = cdist(a,b,'sqeuclidean')


欢迎分享,转载请注明来源:内存溢出

原文地址: http://outofmemory.cn/zaji/5642658.html

(0)
打赏 微信扫一扫 微信扫一扫 支付宝扫一扫 支付宝扫一扫
上一篇 2022-12-16
下一篇 2022-12-16

发表评论

登录后才能评论

评论列表(0条)

保存