PyQtGraph - 获取图形项的 X 轴和 Y 轴数据边界

PyQtGraph 是一个跨平台的 Python 图形库，用于数据可视化。对于数据可视化任务，了解如何获取图形项的 X 轴和 Y 轴数据边界是非常重要的。本文将介绍基本的用法，以及一些针对特定情况的技巧。

获取数据边界的基本用法

在 PyQtGraph 中，每个图形项（例如 PlotItem 或 CurveItem）都有一个 viewBox，表示显示在窗口中的部分。可以使用 viewRect() 方法获取这个矩形的左下角和右上角坐标，这两个坐标即是数据的边界。

from pyqtgraph.Qt import QtGui, QtCore
import pyqtgraph as pg
import numpy as np

app = QtGui.QApplication([])
view = pg.GraphicsView()
layout = pg.GraphicsLayout()
view.setCentralItem(layout)
view.show()

plot = layout.addPlot(title="Example")
x = np.linspace(0, 10, 1000)
y = np.sin(x)
curve = pg.PlotCurveItem(x, y)
plot.addItem(curve)

left, bottom = plot.viewRect().bottomLeft()
right, top = plot.viewRect().topRight()
print(f"X Range: ({left:.2f}, {right:.2f})")
print(f"Y Range: ({bottom:.2f}, {top:.2f})")

app.exec_()

以上代码将创建一个包含正弦曲线的 PlotItem，并打印出 X 轴和 Y 轴数据边界。

针对特定情况的技巧

忽略 NaN 值

如果数据包含 NaN 值，则获取的数据边界通常是不准确的，因为 NaN 值会被视为最大或最小值。可以使用 viewBox().childGroup.childrenBounds() 方法获取忽略 NaN 值后的准确数据边界。

from pyqtgraph.Qt import QtGui, QtCore
import pyqtgraph as pg
import numpy as np

app = QtGui.QApplication([])
view = pg.GraphicsView()
layout = pg.GraphicsLayout()
view.setCentralItem(layout)
view.show()

plot = layout.addPlot(title="Example")
x = np.linspace(0, 10, 1000)
y = np.sin(x)
y[::100] = np.nan  # add some NaN values
curve = pg.PlotCurveItem(x, y)
plot.addItem(curve)

left, bottom = plot.viewBox().childGroup.childrenBounds().bottomLeft()
right, top = plot.viewBox().childGroup.childrenBounds().topRight()
print(f"X Range: ({left:.2f}, {right:.2f})")
print(f"Y Range: ({bottom:.2f}, {top:.2f})")

app.exec_()

以上代码添加了一些 NaN 值，并使用了 childrenBounds() 方法获取的准确数据边界。

获取坐标轴的范围

如果需要获取坐标轴的范围，则可以使用 axis 对象和 range 方法来实现。

from pyqtgraph.Qt import QtGui, QtCore
import pyqtgraph as pg
import numpy as np

app = QtGui.QApplication([])
view = pg.GraphicsView()
layout = pg.GraphicsLayout()
view.setCentralItem(layout)
view.show()

plot = layout.addPlot(title="Example")
x = np.linspace(0, 10, 1000)
y = np.sin(x)
curve = pg.PlotCurveItem(x, y)
plot.addItem(curve)

xAxis = plot.getAxis("bottom")
yAxis = plot.getAxis("left")
print(f"X Axis Range: {xAxis.range}")
print(f"Y Axis Range: {yAxis.range}")

app.exec_()

以上代码将打印出 X 轴和 Y 轴的范围。

结论

获取图形项的 X 轴和 Y 轴数据边界是 PyQtGraph 数据可视化任务的一个重要部分。本文介绍了基本的用法和针对特定情况的技巧，希望能够对读者有所帮助。