PyQt5_pyqtgraph添加点标记

目录

效果：

代码：

使用：

效果：

代码：

import sys
import pandas as pd
from PyQt5 import QtCore,QtWidgets
from PyQt5.QtCore import Qt
from typing import Any,Dict,List
import pyqtgraph as pg
pg.setConfigOption('background', 'w')
pg.setConfigOption('foreground', 'k')

 自定义坐标控件

class RotateAxisItem(pg.AxisItem):
    def drawPicture(self, p, axisSpec, tickSpecs, textSpecs):
        p.setRenderHint(p.Antialiasing,False)
        p.setRenderHint(p.TextAntialiasing,True)

        ## draw long line along axis
        pen,p1,p2 = axisSpec
        p.setPen(pen)
        p.drawLine(p1,p2)
        p.translate(0.5,0)  ## resolves some damn pixel ambiguity

        ## draw ticks
        for pen,p1,p2 in tickSpecs:
            p.setPen(pen)
            p.drawLine(p1,p2)

        ## draw all text
        # if self.tickFont is not None:
        #     p.setFont(self.tickFont)
        p.setPen(self.pen())
        for rect,flags,text in textSpecs:
            # this is the important part
            p.save()
            p.translate(rect.x(),rect.y())
            p.rotate(-30)
            p.drawText(-rect.width(),rect.height(),rect.width(),rect.height(),flags,text)
            # restoring the painter is *required*!!!
            p.restore()

可以添加和去除标记的折线图控件

class PyQtGraphWidget(QtWidgets.QWidget):
    def __init__(self):
        super().__init__()
        self.line_type: int = 1
        self.init_data()
        self.init_ui()
    def init_data(self):
        # 存储标记控件对象
        self.mark_list = []
        # 单根曲线的颜色值
        self.color_one = (30,144,255)
        pass
    def init_ui(self):
        self.title_label = QtWidgets.QLabel('折线图')
        self.title_label.setAlignment(Qt.AlignCenter)
        # 使用自定义的坐标轴，使得横坐标可以倾斜显示
        xax = RotateAxisItem(orientation='bottom')
        xax.setHeight(h=50)
        self.pw = pg.PlotWidget(axisItems={'bottom': xax})
        # 以下两句作用：鼠标滚轮缩放对折线图只作用在x轴上，x,y坐标轴上的数值会跟随缩放
        # 只在x轴缩放折线图可以避免画布中空白，找不到图
        self.pw.setMouseEnabled(x=True,y=False)
        self.pw.setAutoVisible(x=False,y=True)
        layout = QtWidgets.QVBoxLayout()
        layout.addWidget(self.title_label)
        layout.addWidget(self.pw)
        self.setLayout(layout)
        pass

    def set_data(self, data: Dict[str, Any]):
        '''单根y轴'''
        self.line_type = 1
        if data is None:
            self.pw.clear()
            return
        # 将上一次视图清空
        self.pw.clear()
        self.pw.addLegend()
        title_str = data['title_str']
        self.title_label.setText(title_str)
        xTick = [data['xTick00']]
        x = data['x']
        y = data['y']
        y_name = data['y_name']
        self.pw.setLabel('left', y_name)
        self.y_datas = y
        self.x_data = xTick
        self.x_Tick = data['xTick']
        self.y_name = y_name
        xax = self.pw.getAxis('bottom')
        xax.setTicks(xTick)

        self.pw.plot(x, y, connect='finite', pen=pg.mkPen({'color': self.color_one, 'width': 4}), name=self.y_name)

        self.vLine = pg.InfiniteLine(angle=90, movable=False)
        self.hLine = pg.InfiniteLine(angle=0, movable=False)
        self.label = pg.TextItem()

        self.pw.addItem(self.vLine, ignoreBounds=True)
        self.pw.addItem(self.hLine, ignoreBounds=True)
        self.pw.addItem(self.label, ignoreBounds=True)
        self.vb = self.pw.getViewBox()
        self.proxy = pg.SignalProxy(self.pw.scene().sigMouseMoved, rateLimit=60, slot=self.mouseMoved)
        # 显示整条折线图
        self.pw.enableAutoRange()
        pass

    def mouseMoved(self,evt):
        pos = evt[0]
        if self.pw.sceneBoundingRect().contains(pos):
            mousePoint = self.vb.mapSceneToView(pos)
            index = int(mousePoint.x())
            if index >= 0 and index < len(self.y_datas):
                x_str = self.x_Tick[index][1]

                y_str_html = ''
                y_str = str(self.y_datas[index])
                y_str_html += '&nbsp;' + y_str
                if self.line_type == 2:
                    y_str2 = str(self.y_datas2[index])
                    y_str_html += '&nbsp;'+y_str2

                html_str = '<p style="color:black;font-size:18px;font-weight:bold;">&nbsp;' + x_str +'&nbsp;'+y_str_html+ '</p>'
                self.label.setHtml(html_str)
                self.label.setPos(mousePoint.x(), mousePoint.y())
            self.vLine.setPos(mousePoint.x())
            self.hLine.setPos(mousePoint.y())
        pass
    def add_mark(self,data_index_list:List[Any],data_str_list:List[str]):
        '''添加标记'''
        for i,i_item in enumerate(data_index_list):
            temp_str = data_str_list[i]
            targetItem = pg.TargetItem(
                pos=(i_item[0],i_item[1]),
                size=20,
                symbol='star',
                movable=False,
                pen='#FF0000'
            )
            targetItem.setLabel(
                temp_str,
                {
                    "anchor": QtCore.QPointF(0.5, 0.5),
                    "offset": QtCore.QPoint(70, 0),
                    "color": "#FF0000"
                }
            )
            self.mark_list.append(targetItem)
            self.pw.addItem(targetItem)
        pass
    def remove_mark(self):
        '''去除标记'''
        for item in self.mark_list:
            self.pw.removeItem(item)
        pass

使用：

if __name__ == '__main__':
    pre_dir = r'E:/temp055/'
    # twoY start
    df = pd.read_csv(pre_dir +'oneY.csv',encoding='utf-8')
    df['count'] = range(len(df))
    df['val'] = df['val'].astype('float')
    title_str = '添加标记示例'
    x = df['count'].values.tolist()
    x_date = df['date'].values.tolist()
    y = df['val'].values.tolist()
    y_name = '值'
    xTick = []
    for i_x,i_xdate in zip(x,x_date):
        xTick.append((i_x,i_xdate))
    xTick00 = []
    for i in range(0,len(xTick),int(len(xTick)/20)):
        xTick00.append(xTick[i])
    res_data = {
        "x":x,
        "y":y,
        "xTick":xTick,
        "xTick00":xTick00,
        "title_str":title_str,
        "y_name":y_name
    }

    mark_index_list = []
    mark_str_list = []
    # 随机取折线图中的几个点
    for i in [5,10,200,1000,-300,-5]:
        mark_index_list.append([x[i],y[i]])
        temp_str = f"{xTick[i][1]} {y[i]}"
        mark_str_list.append(temp_str)

    QtCore.QCoreApplication.setAttribute(QtCore.Qt.HighDpiScaleFactorRoundingPolicy.PassThrough)
    app = QtWidgets.QApplication(sys.argv)
    temp_widget = PyQtGraphWidget()
    temp_widget.showMaximized()
    temp_widget.set_data(res_data)
    # 添加标记
    temp_widget.add_mark(mark_index_list,mark_str_list)
    # 去除标记
    # temp_widget.remove_mark()
    app.exec()
    # twoY end
    pass