python - 以 .png 格式存储带有颜色的 2D 数组

Question

这个问题已经有答案了:

Saving a Numpy array as an image (23 个回答)

已关闭 6 年前。

我编写了这个 Python 程序来创建矩阵(二维数组)并将其保存到 .png 文件中。程序编译并运行，没有任何错误。即使创建了 IMAGE.png 文件，但该 png 文件也无法打开。当我尝试在 MSPaint 中打开它时，它显示:

Cannot open image. Not a valid bitmap file or its format is not currently supported.

我的目标是根据 2D 数组中存储的数字创建 RBG png 图像。

源代码:

    import numpy;
    import png;

    imagearray = numpy.zeros(shape=(512,512));

    /* Code to insert one '1', '2', '3' in certain locations 
       of the numpy 2D Array. Rest of the location by default stores zero '0'.*/


    f = open("IMAGE.png", 'wb');
    f.write(imagearray);
    f.close();

我不明白我哪里出了问题，因为没有错误消息。请帮忙。

PS-我只想将矩阵保存为图像文件，因此如果您有更好且简单的方法在 Python2.7 中执行此操作，请提出建议。

Answer 1

使用 matplotlib 绘制图像并保存。

import matplotlib.pyplot as plt
import numpy as np
a = np.random.uniform(size=(25, 25, 3))  # random 3D array
plt.imshow(a)
plt.savefig("img.png")

imshow() 有各种感兴趣的参数，其中插值 ( examples here ) 和 cmap (颜色图)

删除轴和空格，按照this question :

plt.axis('off')
plt.savefig("img.png", bbox_inches='tight')