如何在 matplotlib 和 pyqt5 中使用 BytesIO？

Question

我在 matplotlib 中制作了一個圖形，並希望將其制作成圖像並在我的 pyqt5 應用程序中使用它。 有人建議我為此使用 BytesIO。 到目前為止，這是我的代碼：

繪制我的圖形：

...
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)停止工作，我不確定為什么。 另外：我怎么能把它放在我的主窗口中？ 因為我懷疑那里的代碼會起作用，哈哈

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))