我想在Matplotlib中使用imshow
或matshow
创建一个10x10网格。下面的函数将numpy数组作为输入,并绘制网格图。但是,我想从阵列中的值也显示在网格定义的单元格内。到目前为止,我找不到一个正确的方法来做到这一点。我可以使用plt.text
将东西放在网格上,但这需要每个单元的坐标,完全不方便。有没有更好的方式去做我想要完成的事情?Matplotlib imshow/matshow在地图上显示值
谢谢!
注意:下面的代码没有从数组中取值,我只是在玩plt.text
。
import numpy as np
import matplotlib.pyplot as plt
from matplotlib import colors
board = np.zeros((10, 10))
def visBoard(board):
cmap = colors.ListedColormap(['white', 'red'])
bounds=[0,0.5,1]
norm = colors.BoundaryNorm(bounds, cmap.N)
plt.figure(figsize=(4,4))
plt.matshow(board, cmap=cmap, norm=norm, interpolation='none', vmin=0, vmax=1)
plt.xticks(np.arange(0.5,10.5), [])
plt.yticks(np.arange(0.5,10.5), [])
plt.text(-0.1, 0.2, 'x')
plt.text(0.9, 0.2, 'o')
plt.text(1.9, 0.2, 'x')
plt.grid()
visBoard(board)
输出:
使用批注可以让您非常灵活地指定文本的坐标。 – tacaswell
我想知道是否有办法做到这一点,而不指定坐标。到目前为止,我手动完成了它(可能不是最聪明的想法)假设数字大小可能会改变,我将不得不提出一个计算正确坐标的函数。 – marillion