我有以下代码和它生成的图。我的目标是在第二个图(右)上绘制红色平面上的一维高斯分布。
这样做的目的是表明重叠(代表条件)是高斯分布。我对正确分布的确切方差不感兴趣,而只是直观地显示它。
在python中有什么直接的方法可以做到这一点吗?
谢谢,P
import numpy as np
import matplotlib.pyplot as plt
from matplotlib.mlab import bivariate_normal
from mpl_toolkits.mplot3d import Axes3D
#Make a 3D plot
fig = plt.figure(figsize=plt.figaspect(0.5))
################ First Plot ##############
#Parameters to set
mu_x = 0
sigma_x = np.sqrt(5)
mu_y = 0
sigma_y = np.sqrt(5)
#Create grid and multivariate normal
x = np.linspace(-10,10,500)
y = np.linspace(-10,10,500)
X, Y = np.meshgrid(x,y)
Z = bivariate_normal(X,Y,sigma_x,sigma_y,mu_x,mu_y)
# Create plane
x_p = 2
y_p = np.linspace(-10,10,500)
z_p = np.linspace(0,0.02,500)
Y_p, Z_p = np.meshgrid(y_p, …Run Code Online (Sandbox Code Playgroud)