如何将字节与matplotlib和pyqt5一起使用

Question

我在matplotlib中绘制了一个图形，并希望将其放入图像并在我的pyqt5应用程序中使用。有人建议我使用字节来为此。到目前为止，这是我的代码：

绘制我的图形：

...
plt.axis('equal')
buff = io.BytesIO()
plt.savefig(buff, format="png")
print(buff)
return buff

然后在另一个脚本中调用：

def minionRatioGraphSetup(self, recentMinionRatioAvg):
    image = minionRatioGraph(recentMinionRatioAvg)
    label = QtWidgets.QLabel()
    pixmap = QtGui.QPixmap(image)
    label.setPixmap(pixmap)
    label.setGeometry(QtCore.QRect(0,0,200,200))

它停止在pixmap = QtGui.QPixmap(image)工作，我不确定为什么。另外：我该如何将其放在我的Mainwindow中？因为我怀疑那里的代码是否有效

Answer 1

我确定使用缓冲区有一个解决方案。但是，正确的字节格式似乎很复杂。因此，另一种选择是将图像保存到磁盘，然后从那里加载。

import sys
from PyQt4 import QtGui
import matplotlib.pyplot as plt
import numpy as np
def minionRatioGraph():
    plt.plot([1,3,2])
    plt.savefig(__file__+".png", format="png")

class App(QtGui.QWidget):
    def __init__(self):
        super(App, self).__init__()
        self.setGeometry(300, 300, 250, 150)
        self.setLayout(QtGui.QVBoxLayout())
        label = QtGui.QLabel()
        label2 = QtGui.QLabel("Some other text label") 
        minionRatioGraph()
        qimg = QtGui.QImage(__file__+".png")  
        pixmap = QtGui.QPixmap(qimg)
        label.setPixmap(pixmap)
        self.layout().addWidget(label)
        self.layout().addWidget(label2)
        self.show()

if __name__ == '__main__':
    app = QtGui.QApplication([])
    ex = App()
    sys.exit(app.exec_())

Answer 2

使用枕头的摘要，可能有助于避免文件io

 im = PIL.Image.open("filename")
 with BytesIO() as f:
     im.save(f, format='png')
     f.seek(0)
     image_data = f.read()
     qimg = QImage.fromData(image_data)
     patch_qt = QPixmap.fromImage(qimg)

Answer 3

这是一种仅使用缓冲区的解决方案，您需要使用pil创建imageqt，然后将其加载到qpixap

import matplotlib.pyplot as plt
import io
from PIL.ImageQt import ImageQt
from PIL import Image
...
buff = io.BytesIO()
plt.savefig(buff, format="png")
img = Image.open(buff)
img_qt = ImageQt(img)
return img_qt

然后，在您的GUI设置中，您调用您的函数以返回imageqt并使用qpixmap.fromimage((

生成qpixMap

def minionRatioGraphSetup(self, recentMinionRatioAvg):
    image = minionRatioGraph(recentMinionRatioAvg)
    label = QtWidgets.QLabel()
    pixmap = QtGui.QPixmap.fromImage(image)
    label.setPixmap(pixmap)
    label.setGeometry(QtCore.QRect(0,0,200,200))