python - imshow 带有整数轴标签的图

Question

我使用 imshow 图轴上的标签作为不同测试的标签，因此我希望这些标签是真实的。但是我得到的是 float ，例如3.5 ,3.0。我该如何更改？

例子:

import numpy as np
from matplotlib import pyplot as plt
a=np.random.randn(4,4)
plt.imshow(a,extent=[1,5,1,5],interpolation='none')
plt.show()

Answer 1

只需使用整数数组来设置刻度:

plt.xticks(np.arange(1, 6, dtype=np.int))