在seaborn heatmap中划分组的行

Question

我正在用Python绘制海生热图。我的数据本质上是按类别分组的，我想在图上画一些线来表示这些组在地图上的位置。作为一个简单的例子，假设我想从文档中修改这个图。。。

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())