Как сделать 3D-гауссовский сюжет используя numpy?

Question

Я пытаюсь построить гауссовскую функцию, используя numpy.функция z = exp (- (x 2 + y 2) / 10), но я получаю только 2D-функцию

import numpy as np 
from matplotlib import pyplot as plt

x=np.linspace(-10,10, num=100)
y=np.linspace(-10,10, num=100)
z=np.exp(-0.1*x**2-0.1*y**2)

fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')
ax.plot_wireframe(x,y,z)

Я получаю:

но я хочу получить:

Я использую numpy, потому что мне нужен набор данных.

Answer 1

Вам необходимо получить правильные размеры.Это можно сделать с помощью meshgrid.Кроме того, ваш желаемый график - это поверхностный график, а не каркас (хотя вы можете сделать это тоже).

# import for colormaps
from matplotlib import cm

x=np.linspace(-10,10, num=100)
y=np.linspace(-10,10, num=100)

x, y = np.meshgrid(x, y)

z = np.exp(-0.1*x**2-0.1*y**2)

fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')
ax.plot_surface(x,y,z, cmap=cm.jet)
plt.show()