matplotlib.pyplot.matshow()画图

这是一个把矩阵或者数组绘制成图像的函数，使用方法如下：

下面的代码将生成一个12*12的矩阵，每个位置的数据随机生成

import matplotlib.pylab as plt
import numpy as np

def samplemat(dims):
    aa = np.zeros(dims)
    for i in range(dims[0]):
        for j in range(dims[1]):
            aa[i,j] = np.random.randint(1, 10)
    return aa


dimlist = [(12, 12)]
for d in dimlist:
    arr = samplemat(d)
    plt.matshow(arr)

plt.show()

运行结果为：

也可以指定任意行和列的矩阵绘图，下面的代码将画出来一个一行12列的图：

import matplotlib.pylab as plt
import numpy as np

def samplemat(dims):
    aa = np.zeros(dims)
    for i in range(dims[1]):
            aa[0,i] = i
    return aa


dimlist = [(1, 12)]
for d in dimlist:
    arr = samplemat(d)
    plt.matshow(arr)

plt.show()

运行结果：