python - 在 pyqtgraph 中的 x 轴上显示字符串值

Question

我想在 pyqtgraph 中的 x 轴上显示刻度的字符串值。现在我不知道该怎么做。

例如:

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

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

Answer 1

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

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

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

---

可以像这样在 x 轴上绘制自定义字符串。

• 创建一个包含 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_()

示例截图: