如何从numpy 3d mgrid转到位置数组

如何解决如何从numpy 3d mgrid转到位置数组

我正在尝试生成一个多变量高斯函数，该函数将基于3个坐标x，y和z给我输出。我希望每个坐标取一个介于0和199之间的值。

我不确定如何从下面定义的x，y和z到数组大小为200 ^ 3 x 3的数组，该数组包含所有位置或坐标xyz。

我需要一个位置数组，以便可以将它作为scipy multivariate_normal.pdf函数的参数传递。

import numpy as np
from scipy.stats import multivariate_normal

x,y,z= np.mgrid[0:200,0:200,0:200]

mu = np.array([100,100,100])
covar = np.array([[100,0],[0,100]])

解决方法

原来numpy是一个称为vstack的函数，可以完成工作。

import numpy as np
from scipy.stats import multivariate_normal

x,y,z = np.mgrid[0:200,0:200,0:200]

xyz = np.vstack((x.flat,y.flat,z.flat)).T

mu = np.array([100,100,100])
covar = np.array([[100000,0],[0,100000,100000]])

pdf = multivariate_normal.pdf(xyz,mu,covar)

pdf = pdf.reshape(200,200,200)