python matplotlib 绘制3D曲面图

python matplotlib 绘制3D曲面图

""" 绘制3D曲面图 """ import numpy as np import matplotlib.pyplot as mp from mpl_toolkits.mplot3d import Axes3D # 准备数据 n = 1000 x, y = np.meshgrid(np.linspace(-3, 3, n), np.linspace(-3, 3, n)) z = (1 - x / 2 + x ** 5 + y ** 3) * \ np.exp(-x ** 2 - y ** 2) # 绘制图片 fig = mp.figure("3D Surface", facecolor="lightgray") mp.title("3D Surface", fontsize=18) # 设置为3D图片类型 ax3d = Axes3D(fig) # ax3d = mp.gca(projection="3d") # 同样可以实现 ax3d.set_xlabel("X") ax3d.set_ylabel("Y") ax3d.set_zlabel("Z") mp.tick_params(labelsize=10) ax3d.plot_surface(x, y, z, cstride=20, rstride=20, cmap="jet") mp.show()