Color plot auf Matplotlibwidget
Verfasst: Samstag 28. März 2015, 11:30
Hi,
Ich habe bassierend auf meinem Matplotlibwidget aus meiner Library eine zusätliche Funktion Matplotlibwidget.color_plot eingebaut. Diese soll den Betrag einer Komplexen Funktion farbkodiert darstellen. Weder Label noch Graph werden dargestellt. Die .py aus meiner xml:
Mein Matplotlibwidget (abgeändert):
Auf diesem Widget möchte ich eine Funktion plotten:
Was mache ich falsch?
Ich habe bassierend auf meinem Matplotlibwidget aus meiner Library eine zusätliche Funktion Matplotlibwidget.color_plot eingebaut. Diese soll den Betrag einer Komplexen Funktion farbkodiert darstellen. Weder Label noch Graph werden dargestellt. Die .py aus meiner xml:
Code: Alles auswählen
# -*- coding: utf-8 -*-
# Form implementation generated from reading ui file 'interface.ui'
#
# Created: Sat Mar 28 11:05:56 2015
# by: PyQt4 UI code generator 4.9.6
#
# WARNING! All changes made in this file will be lost!
from PyQt4 import QtCore, QtGui
try:
_fromUtf8 = QtCore.QString.fromUtf8
except AttributeError:
def _fromUtf8(s):
return s
try:
_encoding = QtGui.QApplication.UnicodeUTF8
def _translate(context, text, disambig):
return QtGui.QApplication.translate(context, text, disambig, _encoding)
except AttributeError:
def _translate(context, text, disambig):
return QtGui.QApplication.translate(context, text, disambig)
class Ui_MainWindow(object):
def setupUi(self, MainWindow):
MainWindow.setObjectName(_fromUtf8("MainWindow"))
MainWindow.resize(800, 600)
self.centralwidget = QtGui.QWidget(MainWindow)
self.centralwidget.setObjectName(_fromUtf8("centralwidget"))
self.mplwidget = MatplotlibWidget(self.centralwidget)
self.mplwidget.setGeometry(QtCore.QRect(190, 80, 400, 300))
self.mplwidget.setObjectName(_fromUtf8("mplwidget"))
MainWindow.setCentralWidget(self.centralwidget)
self.menubar = QtGui.QMenuBar(MainWindow)
self.menubar.setGeometry(QtCore.QRect(0, 0, 800, 21))
self.menubar.setObjectName(_fromUtf8("menubar"))
MainWindow.setMenuBar(self.menubar)
self.statusbar = QtGui.QStatusBar(MainWindow)
self.statusbar.setObjectName(_fromUtf8("statusbar"))
MainWindow.setStatusBar(self.statusbar)
self.retranslateUi(MainWindow)
QtCore.QMetaObject.connectSlotsByName(MainWindow)
def retranslateUi(self, MainWindow):
MainWindow.setWindowTitle(_translate("MainWindow", "MainWindow", None))
from matplotlibwidget import MatplotlibWidget
Code: Alles auswählen
__version__ = "1.0.0"
from PyQt4.QtGui import QSizePolicy
from PyQt4.QtCore import QSize
from matplotlib.backends.backend_qt4agg import FigureCanvasQTAgg as Canvas
from matplotlib.figure import Figure
from matplotlib import rcParams
rcParams['font.size'] = 9
import numpy as np
class MatplotlibWidget(Canvas):
def __init__(self, parent=None, title='', xlabel='', ylabel='',
xlim=None, ylim=None, xscale='linear', yscale='linear',
width=4, height=3, dpi=100, hold=False):
self.figure = Figure(figsize=(width, height), dpi=dpi)
self.axes = self.figure.add_subplot(111)
self.axes.set_title(title)
self.axes.set_xlabel(xlabel)
self.axes.set_ylabel(ylabel)
if xscale is not None:
self.axes.set_xscale(xscale)
if yscale is not None:
self.axes.set_yscale(yscale)
if xlim is not None:
self.axes.set_xlim(*xlim)
if ylim is not None:
self.axes.set_ylim(*ylim)
self.axes.hold(hold)
Canvas.__init__(self, self.figure)
self.setParent(parent)
Canvas.setSizePolicy(self, QSizePolicy.Expanding, QSizePolicy.Expanding)
Canvas.updateGeometry(self)
def sizeHint(self):
w, h = self.get_width_height()
return QSize(w, h)
def minimumSizeHint(self):
return QSize(10, 10)
def color_plot(self,xval,yval,zval,xlabel='',xunit='',ylabel='',
yunit='',zlabel='',zunit=''):
self.figure.clear()
self.figure.subplots_adjust(bottom=0.15,left=0.17)
self.axes.ticklabel_format(style='sci', axis='x', scilimits=(0,0))
self.axes.ticklabel_format(style='sci', axis='y', scilimits=(0,0))
self.axes.axis([xval.min(),xval.max(),yval.min(),yval.max()])
x_color=self.axes.pcolormesh(xval,yval,np.absolute(zval).T)
self.figure.colorbar(x_color,orientation="vertical",
label=zlabel+' ['+zunit+']',format='%.0e')
self.axes.set_xlabel(xlabel+' ['+xunit+']')
self.axes.set_ylabel(ylabel+' ['+yunit+']')
self.draw()
#===============================================================================
# Example
#===============================================================================
if __name__ == '__main__':
import sys
from PyQt4.QtGui import QMainWindow, QApplication
from numpy import linspace
class ApplicationWindow(QMainWindow):
def __init__(self):
QMainWindow.__init__(self)
self.mplwidget = MatplotlibWidget(self, title='Example',
xlabel='Linear scale',
ylabel='Log scale',
hold=True, yscale='log')
self.mplwidget.setFocus()
self.setCentralWidget(self.mplwidget)
self.plot(self.mplwidget.axes)
def plot(self, axes):
x = linspace(-10, 10)
axes.plot(x, x**2)
axes.plot(x, x**3)
app = QApplication(sys.argv)
win = ApplicationWindow()
win.show()
sys.exit(app.exec_())
Code: Alles auswählen
import sys
from PyQt4.QtCore import *
from PyQt4.QtGui import *
from PyQt4.uic import *
import numpy as np
import plot_options as po
application = QApplication(sys.argv)
frame=loadUi('interface.ui')
def f(x,y):
return (x**2+y**2)*np.exp(np.complex(x,y))
x=np.array([-2.0,-1.0,0.0,1.0,2.0])
y=np.array([-2.0,-1.0,0.0,1.0,2.0])
z=[]
for xi in x:
z_temp=[]
for yi in y:
z_temp.append(f(xi,yi))
z.append(z_temp)
z=np.array(z)
frame.mplwidget.color_plot(x,y,z,xlabel='x',
xunit='-',ylabel='y',
yunit='-',zlabel='z',zunit='-')
frame.show()
application.exec_()