如何将自定义AxisItem添加到现有PlotWidget?

时间:2017-03-19 12:55:28

标签: python python-2.7 pyqt pyqt4 pyqtgraph

我正在尝试将pyqtgraph中的自定义 AxisItem 添加到由Qt Designer生成的现有 PlotWidget 中。有相关主题here,但代码示例没有确切答案,我无法发表评论,因此我创建了一个新主题。

这是我的自定义 AxisItem (基于this代码):

import pyqtgraph as pg
import datetime

def int2td(ts):
    return(datetime.timedelta(seconds=float(ts)/1e6))

class TimeAxisItem(pg.AxisItem):
    def __init__(self, *args, **kwargs):
        super(TimeAxisItem, self).__init__(*args, **kwargs)
    def tickStrings(self, values, scale, spacing):
        return [int2dt(value).strftime("%H:%M:%S") for value in values]

这是我的主要 QtPlotter 类:

from pyqtgraph.Qt import QtGui
from template_pyqt import Ui_Form # Ui_Form is generated by Qt Designer

class QtPlotter:
    def __init__(self):
        self.app = QtGui.QApplication([])
        self.win = QtGui.QWidget()
        self.ui = Ui_Form()
        self.ui.setupUi(self.win)
        self.win.show()

        self.ui_plot = self.ui.plot 
        self.ui_plot.showGrid(x=True, y=True)

然后我尝试添加自定义 AxisItem

self.ui_plot.getPlotItem().axes['bottom']['item'] = TimeAxisItem(orientation='bottom')

我没有错误,但这没有任何效果。

5 个答案:

答案 0 :(得分:2)

PlotItem类没有setAxis方法,只有getAxis方法来获取当前轴。您可以在axisItems构造函数的PlotItem参数中指定包含轴项的字典,但似乎无法更新现有AxisItem的{​​{1}}。< / p>

即使PlotItem字典是一个公共属性,它也没有文档,因此只是使用它是有点冒险的。 PyQtGraph的作者可能会改变它的行为,或者重命名它(虽然这种情况的变化很小)。恕我直言,它应该是一个私人属性。

在任何情况下,通过查看PlotItem.axes构造函数的source,您可以看到如何将PlotItem参数中的轴项添加到绘图项中:

axisItem

也许你可以通过查看上面的代码来使它工作。再次,这是无证行为,使用风险自负!此外,在设置新 ## Create and place axis items if axisItems is None: axisItems = {} self.axes = {} for k, pos in (('top', (1,1)), ('bottom', (3,1)), ('left', (2,0)), ('right', (2,2))): if k in axisItems: axis = axisItems[k] else: axis = AxisItem(orientation=k, parent=self) axis.linkToView(self.vb) self.axes[k] = {'item': axis, 'pos': pos} self.layout.addItem(axis, *pos) axis.setZValue(-1000) axis.setFlag(axis.ItemNegativeZStacksBehindParent) 以防止内存泄漏时,应正确删除和取消链接旧AxisItem。也许这很棘手,这可能是PlotItem.setAxis方法不存在的原因。

答案 1 :(得分:1)

太简单了。 PlotItem 类没有 setAxis 方法。取而代之的是一个名为 setAxisItems 的方法。这段代码对我有用。

date_axis = pg.graphicsItems.DateAxisItem.DateAxisItem(orientation = 'bottom')
self.mainSoundPlot.setAxisItems(axisItems = {'bottom': date_axis})

答案 2 :(得分:0)

我花了整整一天时间研究PlotItem构造函数的source,但无法正常工作。

最后,我发现QtDesigner / QtCreator在PlotWidget上仅输出三行。与其尝试将TimeAxisItem添加到现有的PlotWidget,不如({绝对更好)是,删除现有的PlotWidget,然后使用TimeAxisItem新建一个,如here所述。

from PyQt5 import QtCore
import pyqtgraph as pg

parent = self.ui_plot.parent()
geom_object = self.ui_plot.frameGeometry()
geometry = QtCore.QRect(geom_object.left(), geom_object.top(), geom_object.width(), geom_object.height())
object_name = self.ui_plot.objectName()

del self.ui_plot

time_axis = TimeAxisItem(orientation='bottom')

self.ui_plot = pg.PlotWidget(parent, axisItems={'bottom': time_axis})
self.ui_plot.setGeometry(geometry)
self.ui_plot.setObjectName(object_name)

答案 3 :(得分:0)

由于有些人试图在这里找到一些东西,所以我将发布我简单但不优雅的解决方法。

用pyuic将Qt Designer生成的模板转换为python模板文件后,我通过替换相应的PlotWidget将自定义AxisItem直接添加到python模板文件中。这一切都可以通过简单的bash脚本完成:

pyuic4 template.ui -o template.py 

match="from PyQt4 import QtCore, QtGui"
insert_class="from timeaxisitem_class import TimeAxisItem"
match_widget="self.plot = PlotWidget(Form)"
insert_timeaxis="self.plot = PlotWidget(Form, axisItems={'bottom': TimeAxisItem(orientation='bottom')})"

file="template.py"
sed -i "s/$match/$match\n$insert_class/" $file
sed -i "s/$match_widget/$insert_timeaxis/" $file

答案 4 :(得分:0)

我知道这是一篇老文章,但是我遇到了类似的问题,并遇到了this gist example中的解决方案,该解决方案是实现attachToPlotItem方法的AxisItem子类。

关于原始帖子,而不是

self.ui_plot.getPlotItem().axes['bottom']['item'] = TimeAxisItem(orientation='bottom')

您在TimeAxisItem子类中需要以下方法(从上面的示例复制):

    def attachToPlotItem(self, plotItem):
        """Add this axis to the given PlotItem
        :param plotItem: (PlotItem)
        """
        self.setParentItem(plotItem)
        viewBox = plotItem.getViewBox()
        self.linkToView(viewBox)
        self._oldAxis = plotItem.axes[self.orientation]['item']
        self._oldAxis.hide()
        plotItem.axes[self.orientation]['item'] = self
        pos = plotItem.axes[self.orientation]['pos']
        plotItem.layout.addItem(self, *pos)
        self.setZValue(-1000)

它甚至保留了原始的AxisItem,以便在需要时可以将其恢复。

然后在您的QtPlotter构造函数中添加

axis = TimeAxisItem(orientation='bottom')
axis.attactToPlotItem(self.ui_plot.getPlotItem())