配置matplotlib colorbar以匹配3D表面值

import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D
from matplotlib import cm

fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')

# 假设有一个数据集data，包含x、y、z三个维度的数据
x = data[:, 0]
y = data[:, 1]
z = data[:, 2]
ax.plot_trisurf(x, y, z, cmap=cm.coolwarm)  # 使用coolwarm颜色映射

# 获取当前图形对象的Axes对象
cax = fig.add_axes([0.85, 0.15, 0.05, 0.7])  # 设置colorbar的位置和大小
fig.colorbar(ax.collections[0], cax=cax)  # 将colorbar添加到图形对象中