绘制带有等值线图投影的 Matplotlib 3D plot_surface
要绘制带有等值线图投影的 3D plot_surface,我们可以使用plot_surface()和contourf()方法。
步骤
设置图形大小并调整子图之间和周围的填充。
使用 numpy 创建x、y、X、Y和Z数据点。
使用figure()方法创建一个新图形或激活现有图形。
将'~.axes.Axes'添加到图形中作为子图排列的一部分,并进行 3D 投影。
使用plot_surface()方法创建曲面图。
使用contourf()方法创建 3D 填充等值线图。
关闭该坐标轴。
使用show()方法显示图形。
代码示例
import numpy as np from matplotlib import pyplot as plt plt.rcParams["figure.figsize"] = [7.50, 3.50] plt.rcParams["figure.autolayout"] = True x = np.arange(-5, 5, 0.1) y = np.arange(-5, 5, 0.1) X, Y = np.meshgrid(x, y) Z = X * np.exp(-X - Y) fig = plt.figure() ax = fig.add_subplot(111, projection='3d') surf = ax.plot_surface(X, Y, Z, rstride=8, cstride=8, alpha=1, cmap="copper") ax.contourf(X, Y, Z, zdir='z', offset=np.min(Z), cmap="plasma") ax.contourf(X, Y, Z, zdir='x', offset=-5, cmap="PiYG_r") ax.contourf(X, Y, Z, zdir='y', offset=5, cmap="PuBuGn") ax.axis('off') plt.show()
输出
广告