使用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
的區域,值為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”,即可完成配置。再次啟動,就能正常顯示了。
-
代碼
+關注
關注
30文章
4753瀏覽量
68368 -
python
+關注
關注
56文章
4783瀏覽量
84473 -
數組
+關注
關注
1文章
416瀏覽量
25913
發布評論請先 登錄
相關推薦
評論