3D-сюжет matplotlib с PyQt4 в Qtabwidget (mplwidget) - PullRequest
Новая
       22

3D-сюжет matplotlib с PyQt4 в Qtabwidget (mplwidget)

1 голос
/ 01 января 2011

Я работаю над своим первым приложением Python, использующим PythonXY (2.6 и Qt4). У меня есть окно с некоторыми кнопками и QtabWidget. Когда я нажимаю кнопку, я хочу нанести трехмерное изображение на определенную вкладку. Мне удалось получить 2D-график, но я не могу получить оси в 3D. Через Qt desiger я использую mplwidget. Во многих примерах я вижу использование figure (), но не могу заставить его работать с mplwidget. (Я не уверен, что это вообще возможно)

Не могли бы вы показать мне пример кода, определяющего трехмерные оси (в QtabWidget) с простым графиком?

Большое спасибо

1 Ответ

1 голос
/ 01 февраля 2011

В приведенном ниже коде вы должны найти то, что вы ищете (по крайней мере с matplotlib 1.0.1 и pyqt 4.8): 

class MplPlot3dCanvas(FigureCanvas):
def __init__(self):
    self.surfs = [] # [{"xx":,"yy:","val:"}]
    self.fig = Figure()
    self.fig.suptitle("this is  the  figure  title",  fontsize=12)
    FigureCanvas.__init__(self, self.fig)
    FigureCanvas.setSizePolicy(self,
        QSizePolicy.Expanding,
        QSizePolicy.Expanding)
    FigureCanvas.updateGeometry(self)
    self.ax = Axes3D(self.fig) # Canvas figure must be created for mouse rotation
    self.ax.set_xlabel('row (m CCD)')
    self.ax.set_ylabel('col (m CCD)')
    self.ax.set_zlabel('Phi (m)')
    self.format_coord_org = self.ax.format_coord
    self.ax.format_coord = self.report_pixel

def report_pixel(self, xd, yd):
    s = self.format_coord_org(xd, yd)
    s = s.replace(",", " ")
    return s

def update_view(self):
    for plt in self.plots:
            # del plt
            self.ax.collections.remove(plt)
        self.plots = []
        for surf in self.surfs:
            plt = self.ax.plot_surface(surf["xx"], surf["yy"], surf["val"], rstride=5, cstride=5, cmap=cm.jet, linewidth=1, antialiased=True)
            self.plots.append(plt)
        self.draw()


class MplPlot3dView(QWidget):
    def __init__(self, parent = None):
       super(MplPlot3dView, self).__init__(parent)
       self.canvas = MplPlot3dCanvas()
       self.toolbar = NavigationToolbar(self.canvas, self.canvas)
       self.vbox = QVBoxLayout()
       self.vbox.addWidget(self.canvas)
       self.vbox.addWidget(self.toolbar)
       self.setLayout(self.vbox)
       self.to_update = False

Затем вам просто нужно добавить QWidget в ваш QtaWidgetпродвиньте его в новый класс MplPlot3dView (вы легко найдете ресурс в сети о том, как это сделать).

...