matplotlib matshow标签

Question

我一个月前开始使用matplotlib，所以我还在学习 我正在尝试用matshow做热图。我的代码如下：

data = numpy.array(a).reshape(4, 4)  
cax = ax.matshow(data, interpolation='nearest', cmap=cm.get_cmap('PuBu'), norm=LogNorm())  
cbar = fig.colorbar(cax)

ax.set_xticklabels(alpha)  
ax.set_yticklabels(alpha)

其中alpha是django中带有4个字段的模型：'ABC'，'DEF'，'GHI'，'JKL'

问题是我不知道为什么，标签'ABC'没有出现，留下最后一个没有标签的单元格。
如果有人想知道如何以某种方式修改我的脚本以显示'ABC'，我将不胜感激：）

Answer 1

正在发生的事情是，当使用matshow时，xticks实际上扩展到显示的图形之外。 （我不太确定为什么会这样。但我几乎从未使用过matshow。）

为了证明这一点，请查看ax.get_xticks()的输出。在你的情况下，它是array([-1., 0., 1., 2., 3., 4.])。因此，当您设置xtick标签时，“ABC”位于＆lt; -1，-1＆gt;，并且不会显示在图上。

最简单的解决方案就是在您的标签列表中添加空白标签，例如

ax.set_xticklabels(['']+alpha)
ax.set_yticklabels(['']+alpha)

作为一个完整的例子：

import numpy as np
import matplotlib.pyplot as plt

alpha = ['ABC', 'DEF', 'GHI', 'JKL']

data = np.random.random((4,4))

fig = plt.figure()
ax = fig.add_subplot(111)
cax = ax.matshow(data, interpolation='nearest')
fig.colorbar(cax)

ax.set_xticklabels(['']+alpha)
ax.set_yticklabels(['']+alpha)

plt.show()

Answer 2

另一种方法是指定刻度，然后设置相应的标签。然后，您不必担心多余的刻度。这在很多matplotlib演示中都有介绍。因此，在这里，请注意额外的调用ax.set_xticks和ax.set_yticks

import numpy as np
import matplotlib.pyplot as plt

alpha = ['ABC', 'DEF', 'GHI', 'JKL']

data = np.random.random((4,4))

fig = plt.figure()
ax = fig.add_subplot(111)
cax = ax.matshow(data, interpolation='nearest')
fig.colorbar(cax)

xaxis = np.arange(len(alpha))
ax.set_xticks(xaxis)
ax.set_yticks(xaxis)
ax.set_xticklabels(alpha)
ax.set_yticklabels(alpha)

plt.show()