在上两篇文章中,我们介绍了:
本篇文章将介绍使用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的索引,生成一个包含索引序号和索引值元组的列表: