从numpy数组更新QpushButton图标不起作用

Question

我正在尝试用matplotlib imshow显示的numpy数组制作一个图标。

我成功地将matplotlib的颜色变成了一个numpy维数（n * n * 4）

然后我将这个numpy数组转换为Qimage，然后转换为Qpixmap，以便更新Qpushbutton的Icon。

但是，按钮的图标未设置为我创建的图像。实际上它没有做任何事情。如果我使用硬盘驱动器中的图像，则图标会正确更新。

这是一个代码示例：

from PyQt5.QtGui import *
from PyQt5.QtWidgets import  *
from PyQt5.QtCore import *
import numpy as np
import sys
import matplotlib.pyplot as plt 

class StimEdit(QMainWindow):
    def __init__(self, parent=None):
        super(StimEdit, self).__init__()
        self.parent = parent

        self.centralWidget = QWidget()
        self.color = self.centralWidget.palette().color(QPalette.Background)
        self.setCentralWidget(self.centralWidget)
        self.mainHBOX_param_scene = QHBoxLayout()

        self.B = QPushButton('')
        self.B.setFixedSize(100,100)
        self.B.clicked.connect(self.updateicon)
        self.mainHBOX_param_scene.addWidget(self.B)
        self.centralWidget.setLayout(self.mainHBOX_param_scene)


    def updateicon(self):
        CM = np.random.random((10,10))
        ax = plt.imshow(CM)
        colours = (ax.cmap(ax.norm( CM )) * 255).astype(np.uint8)
        ncols, nrows, ncolors = colours.shape
        # image = QImage(colours.tostring(),ncols, nrows, ncolors,QImage.Format_RGBA8888)
        image = QImage(colours[:,:,:3].tostring(),ncols, nrows, ncolors-1,QImage.Format_RGB888)
        rMyIcon = QPixmap(image)
        self.B.setIcon(QIcon(rMyIcon))
        self.B.setIconSize(QSize(100, 100))
        self.parent.processEvents()


def main():
    app = QApplication(sys.argv)
    ex = StimEdit(app)
    ex.show()
    sys.exit(app.exec_( ))


if __name__ == '__main__':
    main()

Answer 1

你必须复制numpy数组并使用previous answer得到以下内容

def updateicon(self):
    size = QSize(100, 100)
    CM = np.random.random((10,10))
    ax = plt.imshow(CM)
    colours = (ax.cmap(ax.norm( CM )) * 255).astype(np.uint8)
    im = colours[:, :, :3].copy()
    image = QImage(im.data, im.shape[1], im.shape[0], im.strides[0], QImage.Format_RGB888)
    pixmap = QPixmap(image)
    self.B.setIcon(QIcon(pixmap.scaled(size)))
    self.B.setIconSize(size)