如何绘制正态分布的 3d 图

Question

我试图在这个图中绘制类似的东西，但没有成功。

这可能是正态分布的 3d 图。任何人都可以帮助我如何获得这样的情节吗？ 这是我正在尝试的代码。

fig=plt.figure(figsize=(10,6))
ax=plt.axes(projection='3d')
x=np.arange(0,251,0.85)
y=np.arange(0,12,0.85)
X,Y=np.meshgrid(x,y)
for i in range(X.shape[0]):
    pos=np.empty(X.shape)
    R=(pos-0.85)/0.029
    Z=(1/0.29*np.sqrt(2*np.pi))*np.exp(-.5*R**2) #normal distribution
    ax.plot_surface(X,Y,Z) 

Answer 1

这是一个创建看起来有点相似的情节的想法。从 300 个随机中心开始，以非常小的带宽计算一个kde kernel。 gaussian_kde 函数将高斯钟形放置在每个随机中心上并将它们相加。

下面的代码有点慢，因为表面是在 200x200 位置上计算和绘制的。为了缩小z轴，借用this post的代码。

import matplotlib.pyplot as plt
from scipy.stats import gaussian_kde
import numpy as np

x = np.linspace(0, 251, 200)
y = np.linspace(0, 12, 200)
centers_x = np.random.uniform(x[0], x[-1], 300)
centers_y = np.random.uniform(y[0], y[-1], 300)
centers = np.vstack([centers_x, centers_y])
kernel = gaussian_kde(centers, bw_method=0.09)

X, Y = np.meshgrid(x, y)
positions = np.vstack([X.ravel(), Y.ravel()])
Z = np.reshape(kernel(positions).T, X.shape)

fig = plt.figure(figsize=(10, 6))
ax = plt.axes(projection='3d')
surf = ax.plot_surface(X, Y, Z, cmap="hot", linewidth=0, rcount=200, ccount=200)
ax.get_proj = lambda: np.dot(Axes3D.get_proj(ax), np.diag([1, 1, 0.1, 1]))
ax.set_zticks([])
plt.show()