主页 / 问题 / 1430391
Accepted
Konstantin
Asked:2022-07-17 05:07:15 +0000 UTC2022-07-17 05:07:15 +0000 UTC

在 Qimage 上放置颜色数组

  • 772

如何放置一组颜色QImage？

我这样做：

import sys
from PySide2.QtGui import QPixmap,  QImage
from PySide2.QtWidgets import QMainWindow, QApplication, QLabel, QHBoxLayout, QPushButton, QWidget
import numpy as np

# 1. Случайные числа в матрице N на M
N = 5
M = 5
data = np.random.random((N, M))
bins = np.array([0.00, 0.10, 0.15, 0.25, 0.30, 0.40, 0.6, 0.8, 0.95])
palette=np.array([[0, 30, 255], [0, 80, 255],[0, 100, 255],
                  [0, 255, 255], [20, 255, 255],[100, 100, 255],
                  [200, 50, 255], [255, 255, 100],[255, 255, 0]])


class MainWindow(QMainWindow):

    def __init__(self):
        super(MainWindow, self).__init__()
        self.title = "Image Viewer"
        self.setWindowTitle(self.title)
        central_widget = QWidget()
        box = QHBoxLayout()
        self.label = QLabel()
        self.button = QPushButton("Click me")
        self.button.clicked.connect(self.save_heat_map)
        box.addWidget(self.label)
        box.addWidget(self.button)
        central_widget.setLayout(box)
        self.setCentralWidget(central_widget)

    def save_heat_map(self):
        digitize = np.digitize(data, bins, right=True)
        digitize = np.expand_dims(digitize, axis=2)
        im = np.choose(digitize, palette, mode='clip')

        print(im)
        im = np.uint8(im)

        heat_map =  QPixmap(QImage(im, im.shape[1], im.shape[0], QImage.Format_RGB888))
        heat_map.save('test_output.png')


if __name__ == '__main__':
    app = QApplication(sys.argv)
    w = MainWindow()
    w.show()
    sys.exit(app.exec_())

结果，我得到了一张图片，其中第一行颜色是正常的，而第二行和随后的颜色好像有偏移......我不知道为什么 =(

在此处输入图像描述

那些。第二行第一个像素的颜色应该(255, 255, 0)是(255, 0, 255)...

python python-3.x

1 个回答

  1. Best Answer
    import sys
from PySide2.QtGui import QPixmap, QImage
from PySide2.QtWidgets import QMainWindow, QApplication, QLabel, QHBoxLayout, QPushButton, QWidget
import numpy as np
 
# 1. Случайные числа в матрице N на M
N = 500
M = 200
data = np.random.random((N, M))
bins = np.array([0.00, 0.10, 0.15, 0.25, 0.30, 0.40, 0.6, 0.8, 0.95])
palette = np.array([[0, 30, 255, 255], [0, 80, 255, 255], [0, 100, 255, 255],
                    [0, 255, 255, 255], [20, 255, 255, 255], [100, 100, 255, 255],
                    [200, 50, 255, 255], [255, 255, 100, 255], [255, 255, 0, 255]], dtype='uint8')
palette[:,[0,2]]=palette[:,[2,0]] #swap channel for Format_ARGB32
 
 
class MainWindow(QMainWindow):
 
    def __init__(self):
        super(MainWindow, self).__init__()
        self.title = "Image Viewer"
        self.setWindowTitle(self.title)
        central_widget = QWidget()
        box = QHBoxLayout()
        self.label = QLabel()
        self.button = QPushButton("Click me")
        self.button.clicked.connect(self.save_heat_map)
        box.addWidget(self.label)
        box.addWidget(self.button)
        central_widget.setLayout(box)
        self.setCentralWidget(central_widget)
 
    def save_heat_map(self):
        digitize = np.digitize(data, bins, right=True)
        digitize = np.expand_dims(digitize, axis=2)
        im = np.choose(digitize, palette, mode='clip')
        h, w, _ = im.shape
        #heat_map = QPixmap(QImage(im, w, h, 4 * w, QImage.Format_ARGB32))
        heat_map=QImage(im, w, h, 4 * w, QImage.Format_ARGB32)
        heat_map.save('test_output.png')
 
 
if __name__ == '__main__':
    app = QApplication.instance()
    if app is None: 
        app = QApplication(sys.argv)
    w = MainWindow()
    w.show()
    app.exec_()
    sys.exit()
    • 1

相关问题