matplotlib imshow绘制相关矩阵时刻度偏移、首个刻度缺失问题求助

问题根因 #

你遇到的刻度偏移、首个刻度丢失、标签顺序错位问题，核心原因是设置刻度标签和刻度定位器的顺序错误：调用set_xticklabels/set_yticklabels时，matplotlib会绑定当前已有的刻度位置，你后续再调整刻度定位器，就会出现标签和位置错位的情况。

修复代码 #

import matplotlib.pyplot as plt
import matplotlib.ticker as ticker

fig1, ax = plt.subplots(1,1)
corr_matrix = X.corr()
heatplot = ax.imshow(corr_matrix, cmap='jet')

# 第一步：先设置刻度定位器，确定每个刻度的准确位置
tick_spacing = 1
ax.xaxis.set_major_locator(ticker.MultipleLocator(tick_spacing))
ax.yaxis.set_major_locator(ticker.MultipleLocator(tick_spacing))

# 第二步：再绑定刻度标签，和上面的位置一一对应
ax.set_xticklabels(corr_matrix.columns)
ax.set_yticklabels(corr_matrix.columns)

# 可选优化：将x轴刻度移到顶部，更符合相关系数矩阵的阅读习惯
ax.xaxis.tick_top()
# 可选优化：旋转x轴标签避免重叠
plt.setp(ax.get_xticklabels(), rotation=45, ha="left", rotation_mode="anchor")

验证说明 #

调整顺序后，刻度位置会先按数据维度完全生成，再绑定对应列名标签，不会再出现首个标签丢失、顺序错位的问题。

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