在 seaborn 热图中分隔组的线

Question

我在 Python 中将数据绘制为 Seaborn 热图。我的数据本质上是按类别分组的，我希望在绘图上有线条来指示这些组在地图上的位置。作为一个简单的例子，假设我想从文档中修改这个图...

import seaborn as sns; sns.set()
flights = sns.load_dataset("flights")
flights = flights.pivot("month", "year", "passengers")
ax = sns.heatmap(flights, cbar=False)

我想通过绘制如下图来强调一年中各季度之间的比较；我该怎么做？

Answer 1

你想要ax.hlines:

ax.hlines([3, 6, 9], *ax.get_xlim())

在 seaborn 热图中分隔组的线

Lines to separate groups in seaborn heatmap

python

matplotlib

heatmap

seaborn