概述
import numpy as np from scipy.spatial.distance import cdist x = np.array([[[1,2,3,4,5],[5,6,7,8,5]],[[11,22,23,24,[25,26,27,28,5]]]) i,j,k = x.shape xx = x.reshape(i,j*k).T y = np.array([[[31,32,33,34,[35,36,37,38,[[41,42,43,44,[45,46,47,48,5]]]) yy = y.reshape(i,j*k).T results = cdist(xx,yy,'mahalanobis') results = np.diag(results) print results [ 2.28765854 2.75165028 2.75165028 2.75165028 0. 2.75165028 2.75165028 2.75165028 2.75165028 0. 0. 0. 0. 0. 0. ]
我的试用版：
VI = np.linalg.inv(np.cov(xx,yy)) print np.sqrt(np.dot(np.dot((xx-yy),VI),(xx-yy).T))
任何人都可以纠正这种方法吗？
这是它的公式：
X = np.vstack([xx,yy]) V = np.cov(X.T) VI = np.linalg.inv(V) print np.diag(np.sqrt(np.dot(np.dot((xx-yy),(xx-yy).T)))
输出：
[ 2.28765854 2.75165028 2.75165028 2.75165028 0. 2.75165028 2.75165028 2.75165028 2.75165028 0. 0. 0. 0. 0. 0. ]
要在没有在此隐式创建的中间数组的情况下执行此操作,您可能必须牺牲Python循环的C循环：
A = np.dot((xx-yy),VI)
B = (xx-yy).T
n = A.shape[0]
D = np.empty(n)
for i in range(n):
D[i] = np.sqrt(np.sum(A[i] * B[:,i]))编辑：实际上,使用np.einsum voodoo,你可以删除Python循环并加速它(在我的系统上,从84.3μs到2.9μs)：
D = np.sqrt(np.einsum('ij,ji->i',A,B))编辑：正如@Warren Weckesser指出的那样,einsum也可以用来取消中间A和B数组：
delta = xx - yy
D = np.sqrt(np.einsum('nj,jk,nk->n',delta,VI,delta))总结
以上是编程之家为你收集整理的python – 仅使用NumPy计算马哈拉诺比斯距离全部内容，希望文章能够帮你解决python – 仅使用NumPy计算马哈拉诺比斯距离所遇到的程序开发问题。
如果您也喜欢它,动动您的小指点个赞吧
602392714
清零编程群