使用 Seaborn 和 Matplotlib 在热图和线图的共享子图中对齐 x 轴刻度

Posted 2023-02-16

【中文标题】使用 Seaborn 和 Matplotlib 在热图和线图的共享子图中对齐 x 轴刻度

【英文标题】：Align x-axis ticks in shared subplots of heatmap and line plots using Seaborn and Matplotlib

【发布时间】：2020-07-17 22:09:28

【问题描述】：

使用具有共享 x 轴的 Seaborn 绘制 热图和线图，热图的刻度位于热图条的中间。

因此，底部线图将继承热图刻度位置和标签，不反映真实数据，因为线图刻度应从零开始。

换句话说，我需要将两个图的刻度移动到从 x 轴原点开始（最佳），或者将线图向右移动热图单元宽度的一半，保持刻度位置和标签（hacky）。

下面的代码可以快速重现问题：

f,[ax_heat,ax_line]=plt.subplots(nrows=2,figsize=(10, 8),sharex=True)

data_heat = np.random.rand(4, 6)
data_line= np.random.randn(6,1)

sb.heatmap(data=data_heat,robust=True, center=0,cbar=False, ax=ax_heat)
sb.lineplot(data=data_line, ax=ax_line)

【参考方案1】：

这是一个 hacky 解决方案，但您可以将 x 轴向左移动宽度的一半：

import numpy as np
import matplotlib.pyplot as plt
import seaborn as sb

f,[ax_heat,ax_line]=plt.subplots(nrows=2,figsize=(10, 8),sharex=True)

data_heat = np.random.rand(4, 6)
data_line = np.random.randn(6,1)

# generalizable code regardless of spacing:
ax = sb.heatmap(data=data_heat,robust=True, center=0,cbar=False, ax=ax_heat)
width = ax.get_xticks()[1] - ax.get_xticks()[0]
new_ax = ax.get_xticks() - 0.5*width
ax.set_xticks(new_ax)
sb.lineplot(data=data_line, ax=ax_line)
plt.show()

【讨论】：

谢谢 Derek O，正是我需要的！

【参考方案2】：

要将两个图的刻度从 x 轴原点开始，只需在代码末尾添加以下行：

plt.xticks(plt.xticks()[0] - 0.5)

解释：

plt.xticks() 返回 x-tick 位置和标签，因此我们可以通过使用[0] 进行索引来访问这些位置。原来这只是一个连续整数值的列表，所以我们可以通过减去 0.5 将它们向左移动半级。

（部分抄自我对another question的回答）

【讨论】：

谢谢阿恩。这对于连续整数非常有效，就像我提供的示例一样。我错过了指定我需要更大的灵活性以防间隔 ox x-ticks 发生变化。由于 width 参数，@Derek O 的答案更有效。我的不好我没有指定，否则你的解决方案非常优雅。

别担心，我同意 Derek 的更通用的解决方案更好。