如何深入理解Matplotlib3D 绘图函数plot_surface

# -*- coding: utf-8 -*-# author:           inspurer(月小水长)# pc_type           lenovo# create_date:      2019/5/25# file_name:        3DTest# github            https://github.com/inspurer# 微信公众号         月小水长(ID: inspurer)"""绘制3d图形"""import matplotlib.pyplot as pltimport numpy as npfrom mpl_toolkits.mplot3d import Axes3D# 定义figurefig = plt.figure()# 创建3d图形的两种方式# 将figure变为3dax = Axes3D(fig)#ax = fig.add_subplot(111, projection='3d')# 定义x, yx = np.arange(-4, 4, 0.25)y = np.arange(-4, 4, 0.25)# 生成网格数据X, Y = np.meshgrid(x, y)# 计算每个点对的长度R = np.sqrt(X ** 2 + Y ** 2)# 计算Z轴的高度Z = np.sin(R)# 绘制3D曲面# rstride:行之间的跨度  cstride:列之间的跨度# rcount:设置间隔个数，默认50个，ccount:列的间隔个数  不能与上面两个参数同时出现# cmap是颜色映射表# from matplotlib import cm# ax.plot_surface(X, Y, Z, rstride = 1, cstride = 1, cmap = cm.coolwarm)# cmap = "rainbow" 亦可# 我的理解的 改变cmap参数可以控制三维曲面的颜色组合, 一般我们见到的三维曲面就是 rainbow 的# 你也可以修改 rainbow 为 coolwarm, 验证我的结论ax.plot_surface(X, Y, Z, rstride = 1, cstride = 1, cmap = plt.get_cmap('rainbow'))# 绘制从3D曲面到底部的投影,zdir 可选 'z'|'x'|'y'| 分别表示投影到z,x,y平面# zdir = 'z', offset = -2 表示投影到z = -2上ax.contour(X, Y, Z, zdir = 'z', offset = -2, cmap = plt.get_cmap('rainbow'))# 设置z轴的维度，x,y类似ax.set_zlim(-2, 2)plt.show()