试图绘制多元Gaussian dist。在3D图中，matplotlib返回一个空图

Question

我试图通过matplotlib绘制高斯分布，但我得到的只是一个空的数字：

当我搜索互联网时，我理解ax.plot_surface()函数需要三个参数，X值，Y值和Z（从X，Y计算Z的函数）。这是对的吗？

我发布下面的代码，希望你能帮我弄清楚我在做错了什么。谢谢！

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

###############################################
### The multivariate Gaussian density function
###############################################

def pdf_multivariate_gauss(x, mu, cov):
    '''
    Caculate the multivariate normal density (pdf)

    Keyword arguments:
        x = numpy array of a "d x 1" sample vector
        mu = numpy array of a "d x 1" mean vector
        cov = "numpy array of a d x d" covariance matrix
    '''
    assert(mu.shape[0] > mu.shape[1]), 'mu must be a row vector'
    assert(x.shape[0] > x.shape[1]), 'x must be a row vector'
    assert(cov.shape[0] == cov.shape[1]), 'covariance matrix must be square'
    assert(mu.shape[0] == cov.shape[0]), 'cov_mat and mu_vec must have the same dimensions'
    assert(mu.shape[0] == x.shape[0]), 'mu and x must have the same dimensions'
    part1 = 1 / ( ((2* np.pi)**(len(mu)/2)) * (np.linalg.det(cov)**(1/2)) )
    part2 = (-1/2) * ((x-mu).T.dot(np.linalg.inv(cov))).dot((x-mu))
    return float(part1 * np.exp(part2))

# Test
x = np.array([[0],[0]])
mu  = np.array([[0],[0]])
cov = np.eye(2) 

print(pdf_multivariate_gauss(x, mu, cov))

#prints 0.15915494309189535

###############################################
### The plot
###############################################

mu  = np.array([[0],[0]])
cov = np.eye(2) 

def construct_Z(X, Y, mu, cov):
    Z = []
    for i,j in zip(X,Y):
        x = np.array([i,j]).reshape(2,1)
        Z.append(pdf_multivariate_gauss(x, mu, cov))
    return Z

X = linspace(-5, 5, 200)
Y = linspace(-5, 5, 200)
Z = construct_Z(X, Y, mu, cov)

fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')
ax.plot_surface(X, Y, Z, rstride=1, cstride=1, color='0.8',
                       alpha=0.85, linewidth=1)     
plt.show()

Answer 1

我在matplotlib中没有使用3D绘图的专家，但我相信你的数据错了。

正如您在this tutorial中的源代码中所看到的，您的X,Y和Z数据必须是二维数组。您的X和Y是一维的，而您的Z是一个简单的列表。

尝试使用X, Y = np.meshgrid(X, Y)

将数据重新整形为网格