python绘制三维子图

import matplotlib.pyplot as plt
import numpy as np

创建 1 张画布

fig = plt.figure(figsize=(15,5))
figsize设置画布大小，（宽，高）

#===============

向画布添加子图 1

ax = fig.add_subplot(1, 2, 1, projection=‘3d’)

X = np.arange(0, 224, 1)[::-1]

Y = np.arange(0, 224, 1)[::-1]

网格化数据

X, Y = np.meshgrid(X, Y)

Z = origin_img[:,:,1].astype(np.int32) - cw_img[:,:,1].astype(np.int32)

绘制第 1 张图

ax.plot_surface(X, Y, Z, rstride=1, cstride=1, cmap=‘rainbow’)

#===============

向画布添加子图 2

ax2 = fig.add_subplot(1, 2, 2, projection=‘3d’)

生成子图 2 数据

X = np.arange(-2, 2, 0.1)
Y = np.arange(-2, 2, 0.1)
X, Y = np.meshgrid(X, Y)
Z = np.sqrt(X ** 2 + Y ** 2)

绘制第 2 张图

ax2.plot_surface(X, Y, Z, cmap=plt.cm.winter)

显示图

plt.show()