来自 Pivot 的 Seaborn 热图中的数据顺序

所以我有一个使用seaborn创建的热图

revels = rd.pivot("Flavour", "Packet number", "Contents")

ax = sns.heatmap(revels, annot=True, fmt="d", linewidths=0.4, cmap="YlOrRd")

plt.show()

which produces

然而，有两件事我想做，尽管咨询了seaborn的帮助文件，但我一生都无法弄清楚如何做（http://seaborn.pydata.org/ generated/seaborn.heatmap.html http://seaborn.pydata.org/generated/seaborn.heatmap.html)

我想做的就是以不同的方式订购口味。尽管在文本文件中输入的顺序为橙子、太妃糖、巧克力、麦芽、葡萄干、咖啡，但在绘图时不会以这种方式生成。我尝试编辑 yticklabs，但这只是编辑标签，而不是随之移动数据。

谢谢你的帮助

PS数据如下：

Packet number,Flavour,Contents
1,orange,4
2,orange,3
3,orange,2
...
1,toffee,4
2,toffee,3
3,toffee,3
...
etc.

对于第一个问题，您需要对数据进行排序。您的第一行创建一个数据框，然后您可以使用 sortlevel 方法对其进行排序。

创建数据框：

revels = rd.pivot("Flavour", "Packet number", "Contents")

因为您使用 Flavor 作为索引，所以在添加到热图之前使用 sortlevel 方法：

revels.sort_index(level=0, ascending=True, inplace=True)

这将更改热图中数据的顺序。

这显然提供了升序/降序排序，但如果您需要自定义排序顺序，请尝试此链接：pandas 数据框中的自定义排序 https://stackoverflow.com/questions/13838405/custom-sorting-in-pandas-dataframe.

自定义排序示例

revels.index = pd.CategoricalIndex(revels.index, categories= ["orange", "toffee", "chocolate", "malteser", "raisin", "coffee"])
revels.sort_index(level=0, inplace=True)