PYTHON中的单列热图

问题描述

我的目标是拥有单列热图，但由于某些原因，如果我不使用二维数组，我通常用于热图的代码将无法使用。

vec1 = np.asarray([1,2,3,4,5])
fig, ax = plt.subplots()
plt.imshow(vec1, cmap='jet')

我知道将单列向量显示为热图很奇怪，但对于我的目的来说，这是一个很好的视觉效果。我只想要一列彩色方块，我可以沿着y轴标记，以向人们显示事物的排名列表。

---

解决方案

您可以使用库Seborn来执行此操作。在Seborn中，您可以确定要绘制的特定列。在本例中，这将是您的数组。以下内容应该可以实现您的要求

vec1 = np.asarray([1,2,3,4,5])
fig, ax = plt.subplots()
seaborn.heatmap([vec1])

然后，您只需像在pyplotlib中一样在该热点图上进行格式化。

http://seaborn.pydata.org/generated/seaborn.heatmap.html