Python matplotlib 绘图学习笔记

1、使用

import matplotlib
matplotlib.use('qt5agg')		# 防止出现警告：matplotlib 3.6 之后将弃用 xxx

import matplotlib.pyplot as plt	# 使用库

在每次绘图之前，必须要先建立一张画布

fig = plt.figure()		# 默认编号为 1，如果给定参数则会被确定为编号

之后创建一个坐标系

# 在该画布上只放了一个坐标系，并且是 3D 坐标系
ax1 = fig.add_subplot(111, projection='3d')		

按需画图，之后若要展示图片，需要使用

plt.show()

2、绘制 3D 直方图

ax.bar3d(x, y, z, dx, dy, dz, [color=None], [zsort='average'])

• (x, y, z) 分别是一个数组，代表着每一个直方块的坐标

• (dx, dy, dz) 分别是一个数组，代表每一个直方块的（长，宽，高）

其中一维的坐标数据可以先用 X, Y = np.meshgrid() 获取网格，然后再 flatten('F') 展平（关于 flatten 只要记得，不给参数是行优先，参数为 'F' 是列优先即可）

效果图：

 

3、绘制 3D 曲面图

ax2.plot_surface(X, Y, Z, rstride=1, cstride=1, cmap=plt.get_cmap('rainbow'))

• X, Y, Z 分别是横纵竖坐标的网格表示

• rstride cstride 表示横纵方向的步长

• cmap 表示配色方案

效果图：

后续继续补充