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()