将多元分布投影到二维图？

Question

我想做与this 问题的答案相同的事情，但不是在 MATLAB 中，而是在带有 matplotlib 的 Python 中。到目前为止，我已经用代码完成了 3D 绘图

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

mu_x = 0
mu_y = 0

x = np.linspace(-10,10,500)
y = np.linspace(-10,10,500)
X, Y = np.meshgrid(x,y)
pos = np.empty(X.shape + (2,))
pos[:, :, 0] = X; pos[:, :, 1] = Y
rv = multivariate_normal([mu_x, mu_y], [[1, 0.8], [0.8, 1]])

fig = plt.figure(figsize=(10,5))
ax = fig.gca(projection='3d')
ax.plot_surface(X, Y, rv.pdf(pos),cmap='viridis',linewidth=0)
ax.set_xlabel('X axis')
ax.set_ylabel('Y axis')
ax.set_zlabel('Z axis')
ax.auto_scale_xyz([-10, 10], [-10, 10], [0, 0.5])
plt.show()

但是如何将其投影到 2D 等高线图？我在努力

plt.figure()
CS = plt.contour(X, Y, rv)
plt.clabel(CS, inline=1, fontsize=10)

但显然这是不对的，因为 Z 不是数组类型（我收到错误 TypeError: float() argument must be a string or a number 的行 CS = plt.contour(X, Y, rv)）。如何将多元分布投影到二维等高线图？谢谢！

Answer 1

您需要向contour 提供与向plot_surface 提供相同的数组。

CS = plt.contour(X, Y, rv.pdf(pos))