如何让Matplotlib的imshow生成图像而不被绘制

Question

Matplotlib imshow很好地绘制了一个numpy数组。最好用以下代码说明：

from PIL import Image
import matplotlib.pyplot as plt
import numpy as np
rows, cols = 200, 200
mat = np.zeros ((rows, cols))
for r in range(rows):
    for c in range(cols):
        mat[r, c] = r * c
# Handle with matplotlib
plt.imshow(mat)

它创建的图： ，这是我想要的。我想要的是没有轴的图像，因此通过谷歌搜索，我能够组装此功能：

def create_img (image, w, h):
    fig = plt.figure(figsize=(w, h), frameon=False)
    canvas = FigureCanvas(fig)
    #To make the content fill the whole figure
    ax = plt.Axes(fig, [0., 0., 1., 1.])
    ax.set_axis_off()
    fig.add_axes(ax)
    plt.grid(False)
    ax.imshow(image, aspect='auto', cmap='viridis')
    canvas.draw()
    buf = fig.canvas.tostring_rgb()
    ncols, nrows = fig.canvas.get_width_height()
    a = np.fromstring(buf, dtype=np.uint8).reshape(nrows, ncols, 3)  
    plt.close()
    plt.pause(0.01)
    return Image.fromarray(a)

它从一个numpy矩阵生成图像。而且它几乎不会绘制。绘图一会儿，然后图像关闭。接下来，我可以保存图像，这是所有麻烦的原因。

我想知道是否有更简单的方法可以达到相同的目标。我试图通过在第一个示例的代码后面添加一些语句来使用枕头：

# Handle with pillow.Image
img = Image.fromarray(mat, 'RGB')
img.show()
img.save('/home/user/tmp/figure.png')

但这会产生不完整的图像。可能是由于我的一些错误，但我不知道是哪一个。

我不知道如何像imshow其他方式（例如通过枕头）获取具有类似输出的numpy数组的图像。有人知道我如何以与matplotlibs相同的方式从numpy矩阵生成图像，imshow()而不会出现闪烁的图吗？并且以比我构想的create_img函数更简单的方式？

Answer 1

这个简单的案例可能最好由处理plt.imsave。

import matplotlib.pyplot as plt
import numpy as np

rows, cols = 200, 200
r,c = np.meshgrid(np.arange(rows), np.arange(cols))
mat = r*c

# saving as image
plt.imsave("output.png", mat)
# or in some other format
# plt.imsave("output.jpg", mat, format="jpg")