PyQtGraph - 获取折线图中线条的数据边界

在使用PyQtGraph绘制折线图时，有时需要获取绘制的线条所代表的数据边界。这可以帮助我们确定图表的显示区域，以及进行数据的限制和筛选。

下面是一个简单的示例代码，通过获取折线图中的数据边界，并根据其进行图表的缩放，实现了图表局部放大的功能。

import pyqtgraph as pg
import numpy as np
from PyQt5 import QtWidgets, QtGui

class LinePlot(QtWidgets.QWidget):

    def __init__(self):
        super().__init__()

        # generate data
        self.x = np.linspace(0, 10, 100)
        self.y = np.sin(self.x)

        # create plot widget
        self.plot_widget = pg.PlotWidget()
        self.plot_widget.setBackground('w')
        self.plot_widget.plot(self.x, self.y, pen='b')

        # create zoom button
        zoom_btn = QtWidgets.QPushButton('Zoom In')
        zoom_btn.clicked.connect(self.zoom_in)

        # add widgets to layout
        layout = QtWidgets.QVBoxLayout()
        layout.addWidget(self.plot_widget)
        layout.addWidget(zoom_btn)
        self.setLayout(layout)

    def zoom_in(self):
        # get data bounds
        bounds = self.plot_widget.getPlotItem().getViewBox().getViewRange()

        # calculate new axis range
        x_range = (bounds[0][0] - 1, bounds[0][1] + 1)
        y_range = (bounds[1][0] * 1.5, bounds[1][1] * 1.5)

        # set new axis range
        self.plot_widget.setXRange(*x_range)
        self.plot_widget.setYRange(*y_range)

if __name__ == '__main__':
    app = QtWidgets.QApplication([])
    plot = LinePlot()
    plot.show()
    app.exec_()

在这段代码中，我们首先生成了一个简单的正弦函数数据，并通过pg.PlotWidget绘制出了一张折线图。然后，我们添加了一个缩放按钮zoom_btn，并设置了它的点击事件处理函数zoom_in。在zoom_in函数中，我们首先通过getPlotItem().getViewBox().getViewRange()获取了当前绘图区域的数据边界。然后，我们根据这些数据边界，计算出了新的x轴和y轴的取值范围（在原有的基础上进行了微小的改动），并通过setXRange和setYRange方法设置了新的显示范围，完成了局部放大的操作。

需要注意的是，在PyQtGraph中，每个PlotWidget都包含一个plotItem，该plotItem包含了整个绘图区域和所有绘制的图形。我们可以通过getPlotItem()方法获取plotItem对象，并通过plotItem来访问和修改绘图区域的属性。在本例中，我们通过getPlotItem().getViewBox().getViewRange()方法获取了绘图区域的数据边界，通过setXRange和setYRange方法设置了新的显示范围。

在使用PyQtGraph绘制折线图时，获取数据边界是一个常见的需求。通过本文的示例代码，我们可以看到，PyQtGraph提供了简单而灵活的API来实现这一需求。同时，我们也可以通过PlotItem中的其他方法，例如addItem和removeItem等，来添加或删除其他绘制元素。