python - Matplotlib : display array values with imshow

Question

我正在尝试使用 matplotlib 函数(如 imshow)创建网格。
从这个数组:

[[ 1  8 13 29 17 26 10  4],
[16 25 31  5 21 30 19 15]]

我想将值绘制为颜色和文本值本身(1,2，...)在同一个网格上。这就是我目前所拥有的(我只能绘制与每个值相关的颜色):

from matplotlib import pyplot
import numpy as np

grid = np.array([[1,8,13,29,17,26,10,4],[16,25,31,5,21,30,19,15]])
print 'Here is the array'
print grid

fig1, (ax1, ax2)= pyplot.subplots(2, sharex = True, sharey = False)
ax1.imshow(grid, interpolation ='none', aspect = 'auto')
ax2.imshow(grid, interpolation ='bicubic', aspect = 'auto')
pyplot.show()   

Answer 1

您想遍历 grid 中的值，并使用 ax.text 将标签添加到绘图中。

幸运的是，对于二维数组，numpy 有 ndenumerate ，这使得这很简单:

for (j,i),label in np.ndenumerate(grid):
    ax1.text(i,j,label,ha='center',va='center')
    ax2.text(i,j,label,ha='center',va='center')