使用等高线图投影绘制 Matplotlib 3D plot_surface
matplotlibpythondata visualization
要使用等高线图投影绘制 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()
输出
