pyqtgraph数据可视化3:使用PyQtGraph绘制精美折线图——以上证指数为例

在上两篇文章中,我们介绍了:

在了解了基本的PyQtGraph模块绘制图形功能之后,我们通过几个常用常见的数据可视化图形来演示使用PyQtGraph进行Python数据可视化。

本篇,我们介绍使用PyQtGraph模块绘制一个完整的折线图,通过tushare模块获取上证指数过去两个月的指数波动数据作为数据源。

下面我们分步骤讲解这个折线图形的绘制。

引入相关模块

在本例中,我们需要使用到pyqtgraph模块、numpy模块和tushare模块。

import pyqtgraph as pg
import tushare as ts
import numpy as np

获取数据源

我们使用tushare模块的get_hist_data()方法获取上证指数从2017年10月到2017年12月的历史行情数据:

data = ts.get_hist_data('sh',start='2017-10-01',end='2017-12-01').sort_index()

返回的是一个Pandas的DataFrame数据结构,操作起来很方便。

处理数据源

在获取到上证指数的历史行情数据之后,我们需要对其进行一些处理,以方便其后进行坐标轴刻度文本的设置。

首先,将data的日期索引转换为一个字典:

xdict = dict(enumerate(data.index))

再按5步长来去data的索引,生成一个包含索引序号和索引值元组的列表:

axis_1 = [(i,list(data.index)[i]) for i in range(0,len(data.index),5)]

绘制图形

在稍微处理好数据源之后,我们就可以进行图形绘制了。

首先实例化一个QT实例:

app = pg.QtGui.QApplication([])

接着借助GraphicsWindow()子模块创建一个空的图形窗口,并使用title参数设置了窗口的标题:

win = pg.GraphicsWindow(title='州的先生zmister.com pyqtgraph数据可视化 - 绘制精美折线图')

通过之前创建的字典xdict和列表axis_1,设置图形的X坐标轴刻度文本,orientation参数表示坐标轴的位置:

stringaxis = pg.AxisItem(orientation='bottom')
stringaxis.setTicks([axis_1,xdict.items()])

在窗口中添加一个空的图形,通过axisItems参数指定坐标轴及其内容,并使用title参数设置了图形的标题:

plot = win.addPlot(axisItems={'bottom': stringaxis},title='上证指数 - zmister.com绘制')

在图形中添加一个文本:

label = pg.TextItem()
plot.addItem(label)

设置图形的图例:

plot.addLegend(size=(150,80))

设置图形网格的形式,我们设置显示横线和竖线,并且透明度惟0.5:

plot.showGrid(x=True, y=True, alpha=0.5)

绘制开盘和收盘的指数,pen参数表示线的颜色,name参数可用于图例的显示,symbolBrush用来设置点的颜色:

plot.plot(x=list(xdict.keys()), y=data['open'].values, pen='r', name='开盘指数',symbolBrush=(255,0,0),)
plot.plot(x=list(xdict.keys()), y=data['close'].values, pen='g', name='收盘指数',symbolBrush=(0,255,0))

设置图形的轴标签:

plot.setLabel(axis='left',text='指数')
plot.setLabel(axis='bottom',text='日期')

最后设置十字光标:

vLine = pg.InfiniteLine(angle=90, movable=False,)
hLine = pg.InfiniteLine(angle=0, movable=False,)
plot.addItem(vLine, ignoreBounds=True)
plot.addItem(hLine, ignoreBounds=True)
vb = plot.vb
def mouseMoved(evt):
    pos = evt[0]  ## using signal proxy turns original arguments into a tuple
    if plot.sceneBoundingRect().contains(pos):
        mousePoint = vb.mapSceneToView(pos)
        index = int(mousePoint.x())
        pos_y = int(mousePoint.y())
        print(index)
        if 0 < index < len(data.index):
            print(xdict[index],data['open'][index],data['close'][index])
            label.setHtml("<p style='color:white'>日期:{0}</p><p style='color:white'>开盘:{1}</p><p style='color:white'>收盘:{2}</p>".format(xdict[index],data['open'][index],data['close'][index]))
            label.setPos(mousePoint.x(),mousePoint.y())
        vLine.setPos(mousePoint.x())
        hLine.setPos(mousePoint.y())
proxy = pg.SignalProxy(plot.scene().sigMouseMoved, rateLimit=60, slot=mouseMoved)

再按常例,调用app的exec_()方法即可:

app.exec_()

最终运行程序,绘制出来的图表如下所示:

动图如下所示:

 

是不是很简单
有问题欢迎留言讨论:)

猜你也喜欢

  1. 匿名说道:

    谢谢分享

  2. 头条新闻说道:

    文章不错支持一下吧

  3. 扫地神僧说道:

    你好,老师这个折线图带光标的如何嵌入到PyQt5的界面里面,另外一个可以嵌入,这个怎么嵌入呢?

    1. zmister说道:

      使用plotwidget绘制这个图形,再在pyqt里面嵌入一个widget部件,调用它。

      1. 扫地神僧说道:

        谢谢老师

  4. 扫地神僧说道:

    你好,州的先生老师:
    我现在用plotwidget已经嵌入到pyqt5里面了,但是鼠标一进去就死。以下是代码,麻烦老师指点以下
    import pyqtgraph as pg
    import tushare as ts
    import numpy as np
    from PyQt5 import QtCore, QtGui, QtWidgets
    class Ui_MainWindow(object):
    def setupUi(self, MainWindow):
    MainWindow.setObjectName("MainWindow")
    MainWindow.resize(1200, 600)
    self.centralwidget = QtWidgets.QWidget(MainWindow)

    self.data = ts.get_hist_data('sh',start='2017-10-01',end='2017-12-01').sort_index()
    self.xdict = dict(enumerate(self.data.index))
    self.axis_1 = [(i,list(self.data.index)[i]) for i in range(0,len(self.data.index),5)]
    #K线图
    self.pw = pg.PlotWidget(title='wkc')
    self.stringaxis = pg.AxisItem(orientation='bottom')
    self.stringaxis.setTicks([self.axis_1,self.xdict.items()])
    # self.pw.addItem(axisItem={'bottom':self.stringaxis},)
    self.label = pg.TextItem()
    self.pw.addLegend(size=(150,80))
    self.pw.showGrid(x=True,y=True,alpha=0.5)
    self.pw.plot(x=list(self.xdict.keys()),y=self.data['open'].values,pen='r',name='PlayWKC',symbolBrush=(255,0,0))
    self.pw.plot(x=list(self.xdict.keys()),y=self.data['close'].values,pen='g', name='PlayWKC', symbolBrush=(0, 255, 0))

    self.pw.setLabel(axis='left',text='指数')
    self.pw.setLabel(axis='bottom',text='日期')

    #设置十字光标
    self.vLine = pg.InfiniteLine(angle=90,movable=False)
    self.hLine = pg.InfiniteLine(angle=0,movable=False)

    self.pw.addItem(self.vLine,ignoreBounds=True)
    self.pw.addItem(self.hLine, ignoreBounds=True)
    # self.vb = self.pw.vb
    self.centralwidget.setObjectName("centralwidget")
    self.verticalLayoutWidget = QtWidgets.QWidget(self.centralwidget)
    self.verticalLayoutWidget.setGeometry(QtCore.QRect(0, 0, 1201, 601))
    self.verticalLayoutWidget.setObjectName("verticalLayoutWidget")
    self.verticalLayout = QtWidgets.QVBoxLayout(self.verticalLayoutWidget)
    self.verticalLayout.setContentsMargins(0, 0, 0, 0)
    self.verticalLayout.setObjectName("verticalLayout")
    self.tabWidget = QtWidgets.QTabWidget(self.verticalLayoutWidget)
    self.tabWidget.setObjectName("tabWidget")
    self.tab = QtWidgets.QWidget()
    self.tab.setObjectName("tab")
    self.verticalLayoutWidget_2 = QtWidgets.QWidget(self.tab)
    self.verticalLayoutWidget_2.setGeometry(QtCore.QRect(0, 0, 1191, 571))
    self.verticalLayoutWidget_2.setObjectName("verticalLayoutWidget_2")
    self.verticalLayout_2 = QtWidgets.QVBoxLayout(self.verticalLayoutWidget_2)
    self.verticalLayout_2.setContentsMargins(0, 0, 0, 0)
    self.verticalLayout_2.setObjectName("verticalLayout_2")
    self.verticalLayout_2.addWidget(self.pw)
    self.tabWidget.addTab(self.tab, "")
    self.tab_2 = QtWidgets.QWidget()
    self.tab_2.setObjectName("tab_2")
    self.tabWidget.addTab(self.tab_2, "")
    self.verticalLayout.addWidget(self.tabWidget)
    MainWindow.setCentralWidget(self.centralwidget)
    self.statusbar = QtWidgets.QStatusBar(MainWindow)
    self.statusbar.setObjectName("statusbar")
    MainWindow.setStatusBar(self.statusbar)

    self.retranslateUi(MainWindow)
    QtCore.QMetaObject.connectSlotsByName(MainWindow)

    def retranslateUi(self, MainWindow):
    _translate = QtCore.QCoreApplication.translate
    MainWindow.setWindowTitle(_translate("MainWindow", "MainWindow"))
    self.tabWidget.setTabText(self.tabWidget.indexOf(self.tab), _translate("MainWindow", "Tab 1"))
    self.tabWidget.setTabText(self.tabWidget.indexOf(self.tab_2), _translate("MainWindow", "Tab 2"))
    self.proxy = pg.SignalProxy(self.pw.scene().sigMouseMoved, rateLimit=60, slot=self.mouseMoved)
    def mouseMoved(self,evt):
    pos = evt[0]
    if self.pw.sceneBoundingRect().contains(pos):
    mousePoint = self.pw.mapSceneToView(pos)
    index = int(mousePoint.x())
    pos_y = int(mousePoint.y())
    print(index)
    if 0 < index < len(range(1,10)):
    print(self.xdict[index],self.data['open'][index],self.data['slose'][index])
    self.label.setHtml("日期:{0}开盘:{1}收盘:{2}".format(self.xdict[index],self.data['open'][index],self.data['close'][index]))
    self.label.setPos(mousePoint.x(x),mousePoint.y())

    self.vLine.setPos(mousePoint.x())
    self.hLine.setPos(mousePoint.y())
    if __name__ == '__main__':

    import sys
    app = QtWidgets.QApplication(sys.argv)
    widget = QtWidgets.QMainWindow()
    ui = Ui_MainWindow()
    ui.setupUi(widget)
    widget.show()
    sys.exit(app.exec_())

    1. zmister说道:

      参考一下我这篇文章:http://bxu2713810459.my3w.com/archives/187.html

      1. mrvx说道:

        老师你好,我的项目也需要嵌入一个带有十字光标的wight到QT界面,你给的文章网址打不开了,应该怎么办,跪求老师回复 :surprised: :surprised:

        1. 州的先生说道:

          加我微信

  5. schmidtgan说道:

    想请教一下,如何动态添加新的数据点

  6. 青山说道:

    请问其中vb = p1.vb,是什么意义呢?我在使用过程中报错:NameError: vb
    感谢。

  7. Avatar说道:

    折线图中,例如Y轴坐标为0到10,折线在Y的坐标时,分别用'ASCII里的带圆圈的数字符号标注。
    怎么做?

zmister进行回复 取消回复

邮箱地址不会被公开。