Python GUI教程(十三):在GUI中使用pyqtgraph绘图库

本篇将会涉及:

  • pyqtgraph绘图库
  • 在GUI中集成一个pyqtgraph

pyqtgraph绘图库

在GUI中集成图形工具是很多桌面程序常有的功能,最为普遍的,就是各种投资交易工具中的价格走势图。比如股票的指数走势,股票的K线图等。

在PyQt的应用程序中集成图绘功能,有很多种方案,比如:

  • OpenGL:一个功能强大的二维三维专业图形绘制工具;
  • Matplotlib:Python中很经典的图形绘制模块;
  • PyQtGraph:基于Qt平台,提供交互的数据显示图形绘制。

各种图形绘制方案都有自己的优点,但是论与PyQt程序的融合度,还是当属PyQtGraph。

PyQtGraph用于图形绘制有很多的优势:

  • 拥有丰富的图形种类;
  • 能够快速地实时更新绘图数据;
  • 能够进行交互式的操作;
  • 多种标记功能;
  • ……

PyQtGraph的安装很简单,使用pip就可以直接安装:

pip install pyqtgraph

安装之后直接在Python代码中使用import 引入就可以使用了。

import pyqtgraph as pg

PyQtGraph自带一个丰富的示例模块,我们在命令行中输入:

python -m pyqtgraph.examples

就会自动打开示例手册的GUI程序:

我们可以选择运行一个示例进行运行:

前面我们说了PyQtGraph与PyQt的结合度很高,下面我们来看看在PyQt中如何调用PyQtGraph绘制的图形

在GUI中集成pyqtgraph图形

在PyQtGraph中,有几种绘制图形的方法:

  • pyqtgraph.plot():创建一个新的绘图窗口来显示数据;
  • PlotWidget.plot():将一组新的数据添加到现有的绘图小部件;
  • PlotItem.plot():将一组新的数据添加到现有的绘图小部件;
  • GraphicsLayout.addPlot():添加一个新的图形到一个图形层中;

根据PyQtGraph这几种绘图方式,我们在PyQt中可以在窗口部件中显示图形,也可以在一个新的窗口的显示图形。

下面我们在窗口部件中显示茅台股票近半年来的K线图,作为在PyQt中集成显示PyQtGraph图形的例子。

首先,在我们之前创建的GUI的基础上,新增一个选项卡,在其中放置一个VerticalLayout层布局:

然后,按照惯例,将UI文件转换为Python文件,并新建一个Python文件引入。

根据PyQtGraph的示例,我们新建一个绘制K线图的类CandlestickItem:

class CandlestickItem(pg.GraphicsObject):  
    def __init__(self, data):  
        pg.GraphicsObject.**init**(self)  
        self.data = data ## data must have fields: time, open, close, min, max  
        self.generatePicture()

    def generatePicture(self):
        self.picture = QtGui.QPicture()
        p = QtGui.QPainter(self.picture)
        p.setPen(pg.mkPen('w'))
        w = (self.data[1][0] - self.data[0][0]) / 3.
        for (t, open, close, min, max) in self.data:
            p.drawLine(QtCore.QPointF(t, min), QtCore.QPointF(t, max))
            if open > close:
                p.setBrush(pg.mkBrush('g'))
            else:
                p.setBrush(pg.mkBrush('r'))
            p.drawRect(QtCore.QRectF(t-w, open, w*2, close-open))
        p.end()

    def paint(self, p, *args):
        p.drawPicture(0, 0, self.picture)

    def boundingRect(self):
        return QtCore.QRectF(self.picture.boundingRect())

然后,我们创建一个函数chart(),用于通过tushare模块获取股票的历史数据,并将数据传入之前定义好的用于生成K线图的类:

def chart():
    hist_data = ts.get_hist_data('600519',start='2017-05-01',end='2017-11-24')
    data_list = []
    axis = []
    for dates,row in hist_data.iterrows():
        # 将时间转换为数字
        date_time = datetime.datetime.strptime(dates,'%Y-%m-%d')
        t = date2num(date_time)
        # t = dict(enumerate(datetime))
        open,high,close,low = row[:4]
        datas = (t,open,close,low,high)
        data_list.append(datas)
        axis.append(t)
    # print(axis)
    axis_dict = dict(enumerate(axis))
    item = CandlestickItem(data_list)
    plt = pg.PlotWidget()
    # print(plt.getAxis('bottom'))
    plt.addItem(item)
    plt.showGrid(x=True,y=True)
    return plt

接着,在GUI程序的主类MainWindow()的init(self)中进行调用:

class MainWindow(object):  
    def __init__(self):  
        …………  
        self.ui.verticalLayout_3.addWidget(chart())  
        …………

最后我们运行代码:

这样,PyQtGraph的图形作为一个窗口部件集成到了选项卡中。

PyQtGraph的用法和功能还有很多,大家可以在自带的示例库,官方文档(http://www.pyqtgraph.org/documentation/index.html),github中(https://github.com/pyqtgraph/pyqtgraph)获取到最新最全的使用方法和技巧。

下一篇,我们将介绍在GUI中使用网络和数据库。

猜你也喜欢

  1. 匿名说道:

    牛逼

  2. 明以养正说道:

    教程很好,收获很大。GUI中使用网络和数据库还会更新吗?

    1. zmister说道:

      这个系列还有两到三篇的文章,网络和数据库是其中一部分。

  3. 为啥看不见评论?说道:

    t = date2num(date_time)

    老师,date2num()这个方法是哪个库里的? 我一直找不到:sad:

    1. zmister说道:

      这个是matplotlib的方法,在头部我们引入的:from matplotlib.pylab import date2num

  4. 扫地神僧说道:

    这个怎么带光标,实在是不知道用这种方法怎么和前面《pyqtgraph数据可视化3:使用PyQtGraph绘制精美折线图——以上证指数为例》相结合,望老师指点

  5. 灵动说道:

    收获很大,希望能继续更新,能出本书就更好了。 :rolleyes: 望GUI中使用网络和数据库能更新。谢谢。 :rolleyes:

    1. zmister说道:

      嗯,在准备中,精力有限可能会比较慢

  6. 当影子不再是黑色说道:

    请问茅台的股票数据是怎么获取的

  7. 当影子不再是黑色说道:

    ts,为什么报错
    axis为什么报错
    **init**(self)表示什么意思

    1. 匿名说道:

      那是 —init——是继承了类的属性

    2. 匿名说道:

      是继承类的属性,___init__

  8. yep说道:

    老师axis_dict = dict(enumerate(axis))这里的axis是什么啊

  9. 匿名说道:

    没有百度网盘代码呀

  10. 匿名说道:

    请问有源码么?

发表评论

邮箱地址不会被公开。