可视化—三维图的绘制

官方文档

plot_surface( X, Y, Z, *args, norm=None, vmin=None, vmax=None, lightsource=None, **kwargs)

描述
绘制三维图
参数

1. X, Y, Z ：2D arrays
   数据值
2. rcount, ccount ：int
   每个方向使用的最大样品数量。默认值为50。
3. rstride, cstride ：int
   每个方向的下采样步长。这些参数与rcount, ccount是互斥的。
4. cmap ：Colormap
   颜色图

案例

import matplotlib.pyplot as plt
from matplotlib import cm
import numpy as np
from mpl_toolkits.mplot3d import Axes3D

fig, ax = plt.subplots(subplot_kw={"projection": "3d"})
# fig = plt.figure()
# ax = Axes3D(fig)

# Make data.
X = np.arange(-5, 5, 0.25)
Y = np.arange(-5, 5, 0.25)
X, Y = np.meshgrid(X, Y)
R = np.sqrt(X**2 + Y**2)
Z = np.sin(R)

# 画三维图
surf = ax.plot_surface(X, Y, Z, cmap=cm.coolwarm,
                       linewidth=0, antialiased=False)

# 添加颜色条，匹配值
fig.colorbar(surf, shrink=0.5, aspect=5)
ax.set_xlabel('$X$')
ax.set_ylabel('$Y$')
ax.set_zlabel('$Z$')
plt.show()

修改下采样步长

surf = ax.plot_surface(X, Y, Z, cmap=cm.coolwarm,
                       cstride=2, rstride=2,
                       linewidth=0, antialiased=False)

cstride=2或rstride=2表示x轴或y轴的间隔由原来的0.25变成现在的2*0.25

如果对您有帮助，麻烦点赞关注，这真的对我很重要！！！如果需要互关，请评论或者私信！