如何将numpy数组转换为（并显示）图像？

Question

我因此创建了一个数组：

import numpy as np
data = np.zeros( (512,512,3), dtype=np.uint8)
data[256,256] = [255,0,0]

我要这样做的是在512x512图像的中心显示一个红点。（至少从…开始，我想我可以从那里找出其余的内容）

Answer 1

您可以使用PIL创建（并显示）图像：

from PIL import Image
import numpy as np

w, h = 512, 512
data = np.zeros((h, w, 3), dtype=np.uint8)
data[0:256, 0:256] = [255, 0, 0] # red patch in upper left
img = Image.fromarray(data, 'RGB')
img.save('my.png')
img.show()