plt画网格图_Python绘制3D图

参考上一篇的Phthon画2D图：

转不了弯：Phthon画2D图​zhuanlan.zhihu.com

唯一的区别就是3D图把画笔换成了mplot3d，详细参数见下面的代码块

import matplotlib.pyplot as plt
import numpy as np
# 定义figure
fig = plt.figure()
# 创建3d图形的两种方式
# 1、将figure变为3d
ax = Axes3D(fig)
# 2、ax = fig.add_subplot(221, projection='3d')

# 定义x, y
x = np.arange(-4, 4, 0.26)
y = np.arange(-4, 4, 0.26)

# 生成网格数据，相当于笛卡尔积
X, Y = np.meshgrid(x, y) 
# 计算每个点对的长度
R = np.sqrt(X ** 2 + Y ** 2)
Z = np.sin(R)

# 绘制3D曲面
ax.plot_surface(X, Y, Z, rstride = 1, cstride = 1, cmap = plt.get_cmap('coolwarm'))
# rstride:行之间的跨度  cstride:列之间的跨度
# rcount:设置间隔个数，默认50个，ccount:列的间隔个数  不能与上面两个参数同时出现
# cmap参数可以控制三维曲面的颜色组合, 一般三维曲面就是 rainbow 的，可以使用collwarm
# 你也可以修改 rainbow 为 coolwarm, 验证我的结论

# 底部的投影
ax.contour(X, Y, Z, zdir =