来自枢轴的 seaborn 热图中的数据顺序

Question

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

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

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

plt.show()

产生

尽管查阅了 seaborn 的帮助文件 (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.

Answer 1

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

创建数据框：

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

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

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

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

这显然提供了 ascending/descending 排序，但如果您需要自定义排序顺序，请尝试此 link： 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)

Answer 2

上面的例子可行，但你必须用sort_index

替换sortlevel

即revels.sortlevel(level=0, ascending=True, inplace=True) 变为 revels.sort_index(axis=0, ascending=True, inplace=True)

来自枢轴的 seaborn 热图中的数据顺序

Data order in seaborn heatmap from pivot

python

matplotlib

heatmap

pandas

seaborn