Python將二維數組輸出為圖片

原文鏈接

使用Python讀取二維數組，將二維數組輸出為圖片，并保存在本地。

代碼如下：

# coding=utf8
from PIL import Image
import numpy as np
import imageio
import matplotlib.pyplot as pyplot

a = 300
b = 500
x = 20
y = 20
w = 40
h = 80

# 生成圖片矩陣
def Gener_mat(a, b, x, y, w, h):
    img_mat = np.zeros((a, b), dtype=np.int_)
    for i in range(0, a):
        for j in range(0, b):
            img_mat[i][j] = 0
    for i in range(x, x + w):
        for j in range(y, y + h):
            img_mat[i][j] = 1
    return img_mat


# 輸出圖片
def out_img(data):
    data = (data * 255.0).astype('uint8')  # 轉換數據類型
    new_im = Image.fromarray(data)  # 調用Image庫，數組歸一化

    # 顯示新圖片
    pyplot.imshow(data)
    pyplot.show()

    # 保存圖片到本地
    imageio.imsave('new_img.jpg', new_im)


img_mat = Gener_mat(a, b, x, y, w, h)
out_img(img_mat)

其中 Gener_mat 函數用于生成一個300*500的矩陣，矩陣大部分值為0，在坐標(20, 20)處有一個40*80的區(qū)域，值為1。

矩陣轉為的圖片保存在與代碼同級的目錄下，圖片為：

如果不能正常顯示圖片，出現報錯：

MatplotlibDeprecationWarning: Support for FigureCanvases without a required_interactive_framework attribute was deprecated in Matplotlib 3.6 and will be removed two minor releases later.

是Pycharm設置的問題。點擊菜單欄 File——Setting——Tools——Python Scientific，取消勾選“Show plots in tool window”，然后點擊右下角的“OK”，即可完成配置。再次啟動，就能正常顯示了。