我想在Python中生成高斯分布,其中x和y维度表示位置,而z维度表示特定量的大小。
该分布的最大值为2e6,标准偏差sigma = 0.025。
在MATLAB中,我可以这样:
x1 = linspace(-1,1,30);
x2 = linspace(-1,1,30);
mu = [0,0];
Sigma = [.025,.025];
[X1,X2] = meshgrid(x1,x2);
F = mvnpdf([X1(:) X2(:)],mu,Sigma);
F = 314159.153*reshape(F,length(x2),length(x1));
surf(x1,x2,F);
到目前为止,在Python中,我有:
x = np.linspace(-1,1,30)
y = np.linspace(-1,1,30)
mu = (np.median(x),np.median(y))
sigma = (.025,.025)
有一个Numpy函数numpy.random.multivariate_normal,可以与MATLAB的mvnpdf相同,但是我一直在不理解文档。特别是在获得numpy.random.multivariate_normal所需的协方差矩阵时。
从scipy 0.14开始,您可以使用 scipy.stats.multivariate_normal.pdf()
import numpy as np
from scipy.stats import multivariate_normal
x, y = np.mgrid[-1.0:1.0:30j, -1.0:1.0:30j]
# Need an (N, 2) array of (x, y) pairs.
xy = np.column_stack([x.flat, y.flat])
mu = np.array([0.0, 0.0])
sigma = np.array([.025, .025])
covariance = np.diag(sigma**2)
z = multivariate_normal.pdf(xy, mean=mu, cov=covariance)
# Reshape back to a (30, 30) grid.
z = z.reshape(x.shape)
本文收集自互联网,转载请注明来源。
如有侵权,请联系[email protected] 删除。
我来说两句