将类别轴添加到matplotlib matshow

Question

我有300个项目，分别属于几个类别：

前84个项目：A类

接下来的116个项目：B类

最近100项：C类

每个项目之间都有一个距离，我从中创建了一个300x300的对称距离矩阵。

plt.figure(figsize=(10,10)) #to make the figure larger
plt.matshow(distances, fignum=1)

给我一​​个数字（我们称它为图F），其中x和y轴刻度分别为50,100,150，...

但是，我也想在该图中查看类别标签。

例如，图F的垂直轴上的刻度可以位于具有相应类别标签的位置1、85和201。

或者，在图F的左侧添加一条垂直线段也可以，该垂直线段分为三个部分，每个部分都带有类别标签。 （当然，垂直线段的高度应与图F的高度相同）

如何查看哪些行对应于哪个类别？

Answer 1

就像您说的那样，只需添加以下行即可：

plt.matshow(dist)
plt.vlines([84,200],0,299)
plt.hlines([84,200],0,299)

输出：