在 pyqtgraph 中的 x 轴上显示字符串值答案

【问题标题】：Show string values on x-axis in pyqtgraph在 pyqtgraph 中的 x 轴上显示字符串值
【发布时间】：2015-10-24 20:41:35
【问题描述】：

我想在pyqtgraph 中显示 x 轴上刻度的字符串值。现在我无法弄清楚如何做到这一点。

例如：

x = ['a', 'b', 'c', 'd', 'e', 'f']
y = [1, 2, 3, 4, ,5, 6]
pg.plot(x, y)

当我尝试将字符串数组传递给 x 变量时，它会尝试将其转换为浮动并使用错误消息破坏 GUI。

【问题讨论】：

标签： python qt pyqt pyqt4 pyqtgraph

【解决方案1】：

通常在 pyqtgraph 中处理自定义轴字符串时，人们子类化 AxisItem 并用他们想要显示的字符串覆盖 tickStrings。

参见例如pyqtgraph : how to plot time series (date and time on the x axis)?

Pyqtgraphs axisitem 也有一个内置的setTicks，它允许您指定要显示的刻度，这可以针对像这样的简单问题完成，而不是继承 AxisItem。

可以像这样在 x 轴上使用自定义字符串进行绘图。

创建一个 dict，其中包含 x 值和字符串，以显示在轴上。

xdict = {0:'a', 1:'b', 2:'c', 3:'d', 4:'e', 5:'f'}

或通过使用

x = ['a', 'b', 'c', 'd', 'e', 'f']
xdict = dict(enumerate(x))

在 AxisItem 中使用 setTicks 或 子类 AxisItem 并在 tickStrings 中找到该值对应的字符串。

1。使用标准 pyqtgraph AxisItem 和 setTicks

    from PyQt4 import QtCore
    import pyqtgraph as pg

    x = ['a', 'b', 'c', 'd', 'e', 'f']
    y = [1, 2, 3, 4, 5, 6]
    xdict = dict(enumerate(x))

    win = pg.GraphicsWindow()
    stringaxis = pg.AxisItem(orientation='bottom')
    stringaxis.setTicks([xdict.items()])
    plot = win.addPlot(axisItems={'bottom': stringaxis})
    curve = plot.plot(list(xdict.keys()),y)

    if __name__ == '__main__':
        import sys
        if sys.flags.interactive != 1 or not hasattr(QtCore, 'PYQT_VERSION'):
            pg.QtGui.QApplication.exec_()

2。通过继承 AxisItem 实现

这是一种更通用的方法，可以轻松更改为各种有趣的东西，例如将 unix 时间戳转换为日期。

    from PyQt4 import QtCore
    import pyqtgraph as pg
    import numpy as np

    class MyStringAxis(pg.AxisItem):
        def __init__(self, xdict, *args, **kwargs):
            pg.AxisItem.__init__(self, *args, **kwargs)
            self.x_values = np.asarray(xdict.keys())
            self.x_strings = xdict.values()

        def tickStrings(self, values, scale, spacing):
            strings = []
            for v in values:
                # vs is the original tick value
                vs = v * scale
                # if we have vs in our values, show the string
                # otherwise show nothing
                if vs in self.x_values:
                    # Find the string with x_values closest to vs
                    vstr = self.x_strings[np.abs(self.x_values-vs).argmin()]
                else:
                    vstr = ""
                strings.append(vstr)
            return strings

    x = ['a', 'b', 'c', 'd', 'e', 'f']
    y = [1, 2, 3, 4, 5, 6]
    xdict = dict(enumerate(x))

    win = pg.GraphicsWindow()
    stringaxis = MyStringAxis(xdict, orientation='bottom')
    plot = win.addPlot(axisItems={'bottom': stringaxis})
    curve = plot.plot(list(xdict.keys()),y)

    if __name__ == '__main__':
        import sys
        if sys.flags.interactive != 1 or not hasattr(QtCore, 'PYQT_VERSION'):
            pg.QtGui.QApplication.exec_()

示例截图：

【讨论】：

干得好，作为创建xdict 的更好方法，您可以使用enumerate：xdict = dict(enumerate(x))
你也可以通过使用x = 'abcdef'而不是x = ['a', 'b', 'c', 'd', 'e', 'f']来省去很多繁琐的打字工作

【解决方案2】：

我发现最简单的方法是准备一个索引列表和一个字符串列表，然后将它们 zip 放在一起：

ticks = [list(zip(range(5), ('a', 'b', 'c', 'd', 'e')))]

您可以像这样获取 PlotWidget 的现有 AxisItem：

pw = pg.PlotWidget()
xax = pw.getAxis('bottom')

最后像这样设置轴的刻度：

xax.setTicks(ticks)

据我所知，PlotWidgets 自动包含“底部”和“左侧”轴项，但您可以根据需要创建和添加其他项。

【讨论】：