如何用matplotlib绘制三维高斯分布

Question

我已经获得了三维高斯分布的均值和sigma，然后我想用python代码绘制三维分布，并获得分布图。

感谢。

Answer 1

这是基于mpl_toolkits的文件和基于SciPy的multinormal PDF

import matplotlib.pyplot as plt 
from mpl_toolkits.mplot3d import Axes3D 

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([.5, .5]) 
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) 





fig = plt.figure() 

ax = fig.add_subplot(111, projection='3d') 



ax.plot_surface(x,y,z) 
#ax.plot_wireframe(x,y,z) 

plt.show() 

参考上SO答案： -

1. Generating 3D Gaussian distribution in Python

2. http://matplotlib.org/mpl_toolkits/mplot3d/tutorial.h