从 numpy 数组在 PyQt5 中显示图像

Question

我有一个形状为 numpy 的数组(500, 500, 3)。我希望它转换成图像并使用 PyQt5 显示它。

Answer 1

我假设该数组的类型为 uint8 并代表红色、绿色和蓝色 您可以Pillow为此使用，例如

from PyQt5 import QtWidgets, QtGui

from PIL import Image, ImageQt

import numpy as np

# generate data

table = np.zeros((256,256,3), dtype=np.uint8)

for i in range(256):

    table[:,i,0] = i

    table[i,:,1] = i

table[:,:,2] = (2*255 - table[:,:,0] - table[:,:,1]) // 2

# convert data to QImage using PIL

img = Image.fromarray(table, mode='RGB')

qt_img = ImageQt.ImageQt(img)

app = QtWidgets.QApplication([])

w = QtWidgets.QLabel()

w.setPixmap(QtGui.QPixmap.fromImage(qt_img))

w.show()

app.exec()