目标函数,其中x,y取值范围均在[-2,2]
代码如下:
import numpy as np
from matplotlib import pyplot as plt
from matplotlib import cm
x, y = np.mgrid[-2:2:20j, -2:2:20j]
z = x * np.exp(-x ** 2 - y ** 2)
fig = plt.figure(figsize=(8, 6))
ax = fig.gca(projection='3d')
ax.plot_surface(x,y,z,cmap=cm.BuGn_r)
plt.show()
结果图像