如何让3D绘图中colorbar随plot_surface同步旋转对齐

问题场景 #

在matplotlib的3D绘图中，使用plot_surface绘制的图像需要搭配colorbar，要求3D坐标轴旋转时，colorbar能同步跟随视角旋转。但默认方式添加的colorbar基于2D轴坐标，始终保持固定方向，无法随3D视角变化。

解决方案 #

放弃使用原生colorbar函数，转而将colorbar作为一个3D曲面绘制在主图旁，使其成为3D场景的一部分，自然跟随视角旋转。同时手动添加刻度和标签模拟原生colorbar的效果。

修改后的完整代码 #

import matplotlib.pyplot as plt
import numpy as np

fig = plt.figure(figsize=(10, 6))
ax = fig.add_subplot(111, projection="3d", computed_zorder=False)

# 设置初始视角
ax.view_init(roll=ax.roll-5, elev=ax.elev - 20, azim=-75)

# 主绘图数据网格
xx, yy = np.meshgrid(np.arange(1, 8), np.arange(1, 8))
# 绘制主3D曲面（可根据需求替换facecolors参数）
im = ax.plot_surface(
    xx,
    np.zeros_like(xx),
    yy,
    zorder=1,
    vmin=-1,
    vmax=1,
    cmap="inferno",
    alpha=1,
    zsort="max",
)

# --- 构建可旋转的3D Colorbar ---
# 定义colorbar的3D位置：主图右侧，y轴固定值，x轴范围、z轴范围匹配主图的z轴
cb_x = np.array([[xx.max()+0.5, xx.max()+1], [xx.max()+0.5, xx.max()+1]])
cb_y = np.array([[1.2, 1.2], [1.2, 1.2]])
cb_z = np.array([[yy.min(), yy.min()], [yy.max(), yy.max()]])
# 生成与主图一致的颜色渐变
cb_colors = plt.cm.inferno(np.linspace(0, 1, cb_z.shape[0]))
# 绘制colorbar曲面
cb_surface = ax.plot_surface(cb_x, cb_y, cb_z, facecolors=cb_colors, zorder=2)

# 添加colorbar刻度与标签
tick_vals = [-1, 0, 1]
tick_pos = np.linspace(yy.min(), yy.max(), len(tick_vals))
for val, pos in zip(tick_vals, tick_pos):
    ax.text(xx.max()+1.2, 1.2, pos, str(val), fontsize=10)
# 添加colorbar标题
ax.text(xx.max()+0.75, 1.2, yy.max()+0.5, 'Value', fontsize=12, ha='center')

plt.show()

关键说明 #

1. 3D化Colorbar：通过plot_surface绘制一个窄条曲面作为colorbar，使其成为3D场景的一部分，自然跟随视角旋转。
2. 位置调整：通过cb_x、cb_y、cb_z参数可以灵活调整colorbar的位置、长度和宽度，适配主图布局。
3. 颜色同步：使用与主图相同的colormap生成渐变颜色，确保颜色映射一致。
4. 手动刻度：通过ax.text添加刻度文本和标题，模拟原生colorbar的交互效果。

内容的提问来源于stack exchange，提问作者user19393974