Matplotlib 3D绘图：将颜色条（Colorbar）整合至Z轴的实现方法

这个问题问得好！Z轴刻度和颜色条重复展示同一数值范围确实有点冗余，手动移除Z轴、旋转对齐颜色条的方法又太繁琐。其实Matplotlib里有个简洁的方案，能直接实现Z轴与颜色条的整合，完全不用那些麻烦的操作。

实现思路 #

1. 隐藏原Z轴的刻度与标签：既然我们打算用颜色条来承载Z值的展示，直接关掉默认的Z轴标记即可。
2. 调整颜色条的位置与方向：让颜色条沿着Z轴的垂直方向排布，紧贴3D图表，同时给颜色条设置对应Z值的标签，让它自然替代原Z轴的功能。

修改后的完整代码 #

from mpl_toolkits.mplot3d import Axes3D
import matplotlib.pyplot as plt
from matplotlib import cm
from matplotlib.ticker import LinearLocator, FormatStrFormatter
import numpy as np

fig = plt.figure()
ax = fig.gca(projection='3d')
# 生成数据
X = np.arange(-5, 5, 0.25)
Y = np.arange(-5, 5, 0.25)
X, Y = np.meshgrid(X, Y)
R = np.sqrt(X**2 + Y**2)
Z = np.sin(R)

# 绘制3D曲面
surf = ax.plot_surface(X, Y, Z, cmap=cm.coolwarm, linewidth=0, antialiased=False)

# 自定义Z轴：隐藏刻度与标签
ax.set_zlim(-1.01, 1.01)
ax.set_zticks([])  # 移除Z轴刻度标记
ax.set_zlabel('')  # 移除Z轴标签

# 添加与Z轴对齐的颜色条，替代原Z轴功能
cbar = fig.colorbar(surf, ax=ax, orientation='vertical', pad=0.01)
cbar.set_label('Z 值')  # 设置颜色条标签，对应原Z轴含义
# 给颜色条应用和原Z轴一致的刻度格式，保持视觉统一
cbar.ax.yaxis.set_major_locator(LinearLocator(10))
cbar.ax.yaxis.set_major_formatter(FormatStrFormatter('%.02f'))

plt.show()

代码说明 #

• ax.set_zticks([]) 和 ax.set_zlabel('')：彻底隐藏原Z轴的刻度和标签，消除重复信息。
• orientation='vertical'：让颜色条保持垂直方向，和Z轴的方向完全一致。
• pad=0.01：把颜色条紧贴3D图表，看起来就像是原Z轴被替换成了颜色条，而非独立的附加元素。
• 给颜色条设置和原Z轴一致的刻度格式，保证视觉上的连贯性，不会出现风格脱节的问题。

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