PyQtGraph 绘图
数据绘图方案
Python语言 的数据可视化(绘图) 方法,常见的有 Matplotlib 和 PyQtGraph
- Matplotlib
说到 Python语言 的数据作图, Matplotlib 当然是最有名的。
优点: 功能完备、成熟稳定、社区生态圈庞大。
缺点: 某些作图场景性能不高。
- PyQtGraph
PyQtGraph 是基于Qt 的纯Python 库。
优点: 大数据量的作图性能高于 Matplotlib, 动态更新图的性能也比Matplotlib高。
并且和Qt图形界面框架完美融合,因为它的2D作图就是基于Qt的 Graphics View Framework 开发的。
缺点: 作图功能没有Matplotlib多,开发社区没有Matplotlib大。
那么,我们应该使用哪种方案呢?我的建议是:
如果你已经使用Qt开发图形界面程序了,并且作图功能是PyQtGraph支持的, 建议使用 PyQtGraph,因为它和Qt界面无缝结合。
否则 使用 Matplotlib。
本文先介绍 PyQtGraph 的使用示例。
PyQtGraph 安装
官方文档 和 案例
PyQtGraph 官方文档可以点击这里查阅
其中包含了很多示例代码,我们只需运行如下代码,即可查看这些demo和对应的代码
曲线图 示例
下面是一个最常见的 根据x/y轴对应的值 作曲线图的例子
import pyqtgraph as pg
# 创建 一个绘图控件类 PlotWidget 对象,作为顶层控件
# PlotWidget 间接继承 QtWidgets.QGraphicsView
pw = pg.plot()
# 设置图表标题、颜色、字体大小
pw.setTitle("气温趋势",color='teal',size='12pt')
# 背景色改为白色
pw.setBackground('w')
# 显示表格线
pw.showGrid(x=True, y=True)
# 设置上下左右的label
# 第一个参数 只能是 'left', 'bottom', 'right', or 'top'
pw.setLabel("left", "气温(摄氏度)")
pw.setLabel("bottom", "时间")
# 设置Y轴 刻度 范围
pw.setYRange(min=-10, # 最小值
max=50) # 最大值
# 创建 PlotDataItem ,这种类型是画 曲线 Curve 和 散列点 scatter plots 的
curve = pw.plot( pen=pg.mkPen('b')) # 线条颜色
hour = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
temperature = [30, 32, 34, 32, 33, 31, 29, 32, 35, 45]
curve.setData(
hour, # x坐标
temperature # y坐标
)
pg.Qt.exec_()
清除画图区,重新绘制
如果使用新的数据再次绘图,可以先调用clear方法清除原来的内容(plotitem),如下
# 清除原来的plot内容
pw.clear()
# 创建 PlotDataItem ,缺省是曲线图
curve = pw.plot( pen=pg.mkPen('b')) # 线条颜色
hour = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
temperature = [130, 132, 134, 132, 133,131, 129, 132, 135, 145]
curve.setData(hour, # x坐标
temperature # y坐标
)
PlotWidget 和 GraphicsLayoutWidget
PyQtGraph 中 绘图控件类 有两种 PlotWidget 和 GraphicsLayoutWidget, 都是 GraphicsView 子类。GraphicsView 是 Qt 的 QGraphicsView
子类,在其基础上改进了一些功能。
PlotWidget 只能内置一个 绘图对象PlotItem, 而 GraphicsLayoutWidget 可以内置多个 绘图对象。 通常我们使用最多的是PlotWidget
PlotWidget对象的 内置绘图对象 PlotItem ,可以通过 getPlotItem() 方法获取。
为了方便,大部分的PlotItem方法都可以直接通过 PlotWidget对象调用。 比如我们上面示例代码中的 setTitle、showGrid、setLabel、setYRange、plot 等。
调用 plot方法,会创建一个PlotDataItem, 缺省是曲线图
关于PyQtGraph绘图基本架构的更多细节,点击这里查看官方文档
嵌入到Qt程序界面中
上面的例子,图表是在单独的程序中运行显示。
如果我们要把它 嵌入到我们的Qt程序界面 中,主要通过 pyqtgraph 的 PlotWidget 或者 GraphicsLayoutWidget 控件类, 代码如下所示
from PySide2 import QtWidgets
import pyqtgraph as pg
class MainWindow(QtWidgets.QWidget):
def __init__(self):
super().__init__()
self.setWindowTitle('pyqtgraph作图示例')
# 创建 PlotWidget 对象
self.pw = pg.PlotWidget()
# 设置图表标题
self.pw.setTitle("气温趋势",color='#008080',size='12pt')
# 设置上下左右的label
self.pw.setLabel("left","气温(摄氏度)")
self.pw.setLabel("bottom","时间")
# 背景色改为白色
self.pw.setBackground('w')
hour = [1,2,3,4,5,6,7,8,9,10]
temperature = [30,32,34,32,33,31,29,32,35,45]
# hour 和 temperature 分别是 : x, y 轴上的值
self.pw.plot(hour,
temperature,
pen=pg.mkPen('b') # 线条颜色
)
# 创建其他Qt控件
okButton = QtWidgets.QPushButton("OK")
lineEdit = QtWidgets.QLineEdit('点击信息')
# 水平layout里面放 edit 和 button
hbox = QtWidgets.QHBoxLayout()
hbox.addWidget(lineEdit)
hbox.addWidget(okButton)
# 垂直layout里面放 pyqtgraph图表控件 和 前面的水平layout
vbox = QtWidgets.QVBoxLayout()
vbox.addWidget(self.pw)
vbox.addLayout(hbox)
# 设置全局layout
self.setLayout(vbox)
if __name__ == '__main__':
app = QtWidgets.QApplication()
main = MainWindow()
main.show()
app.exec_()
柱状图
pyqtgraph 可以产生这样的 柱状图
from PySide2 import QtWidgets
import pyqtgraph as pg
class MainWindow(QtWidgets.QWidget):
def __init__(self):
super().__init__()
self.setWindowTitle('pyqtgraph作图示例')
# 创建 PlotWidget 对象
self.pw = pg.PlotWidget()
# 设置图表标题
self.pw.setTitle("订单数量",color='#008080',size='12pt')
# 背景色改为白色
self.pw.setBackground('w')
# 设置上下左右的label
self.pw.setLabel("left", "订单量(条)")
self.pw.setLabel("bottom", "日期")
# 显示表格线
self.pw.showGrid(x=True, y=True)
# 产生两种柱状图数据,分别对应 红色柱 和蓝色柱
x1 = [0, 1, 2, 3, 4, 5, 6, 7, 8, 9]
y1 = [0, 1, 2, 3, 4, 5, 6, 7, 8, 9]
x2 = [0.33, 1.33, 2.33, 3.33, 4.33, 5.33, 6.33, 7.33, 8.33, 9.33]
y2 = [0.33, 1.33, 2.33, 3.33, 4.33, 5.33, 6.33, 7.33, 8.33, 9.33]
bg1 = pg.BarGraphItem(x=x1, height=y1, width=0.3, brush='r')
bg2 = pg.BarGraphItem(x=x2, height=y2, width=0.3, brush='g')
# 添加到界面上
self.pw.addItem(bg1)
self.pw.addItem(bg2)
# 创建其他Qt控件
okButton = QtWidgets.QPushButton("OK")
lineEdit = QtWidgets.QLineEdit('点击信息')
# 水平layout里面放 edit 和 button
hbox = QtWidgets.QHBoxLayout()
hbox.addWidget(lineEdit)
hbox.addWidget(okButton)
# 垂直layout里面放 pyqtgraph图表控件 和 前面的水平layout
vbox = QtWidgets.QVBoxLayout()
vbox.addWidget(self.pw)
vbox.addLayout(hbox)
# 设置全局layout
self.setLayout(vbox)
if __name__ == '__main__':
app = QtWidgets.QApplication()
main = MainWindow()
main.show()
app.exec_()
绘制多个图形
可以使用 GraphicsLayoutWidget,创建多个绘图对形
对应代码如下
本节内容 仅 内部学员 可见
实时更新图
要画动态的实时更新图,只需要在把变更的内容重新plot即可。
示例代码如下
from PySide2 import QtWidgets
from pyqtgraph.Qt import QtCore
import pyqtgraph as pg
import sys
from random import randint
class MainWindow(QtWidgets.QMainWindow):
def __init__(self):
super().__init__()
self.setWindowTitle('pyqtgraph作图')
# 创建 PlotWidget 对象
self.pw = pg.PlotWidget()
# 设置图表标题
self.pw.setTitle("气温趋势",
color='#008080',
size='12pt')
# 设置上下左右的label
self.pw.setLabel("left","气温(摄氏度)")
self.pw.setLabel("bottom","时间")
# 设置Y轴 刻度 范围
self.pw.setYRange(min=-10, # 最小值
max=50) # 最大值
# 显示表格线
self.pw.showGrid(x=True, y=True)
# 背景色改为白色
self.pw.setBackground('w')
# 设置Y轴 刻度 范围
self.pw.setYRange(min=-10, # 最小值
max=50) # 最大值
# 居中显示 PlotWidget
self.setCentralWidget(self.pw)
# 实时显示应该获取 PlotDataItem对象, 调用其setData方法,
# 这样只重新plot该曲线,性能更高
self.curve = self.pw.plot(
pen=pg.mkPen('r', width=1)
)
self.i = 0
self.x = [] # x轴的值
self.y = [] # y轴的值
# 启动定时器,每隔1秒通知刷新一次数据
self.timer = QtCore.QTimer()
self.timer.timeout.connect(self.updateData)
self.timer.start(1000)
def updateData(self):
self.i += 1
self.x.append(self.i)
# 创建随机温度值
self.y.append(randint(10,30))
# plot data: x, y values
self.curve.setData(self.x,self.y)
if __name__ == '__main__':
app = QtWidgets.QApplication()
main = MainWindow()
main.show()
app.exec_()
在Qt Designer中加入第三方控件
PyQtGraph图形可以作为一个 Qt的 widget控件,嵌入到 Qt 程序主窗口中。
我们可以在 Qt Designer 中把 PyQtGraph图形控件 作为第三方控件 加入。
比如,像下面这样:
通过 Qt Designer,我们可以预先把界面上的控件的位置大小设计好,然后动态加载。
但是 界面上摆放的都是 Qt内置的控件, 那么像 PyQtGraph 里面的 PlotWidget这种第三方控件怎么 放到 Qt Designer中呢?
最简单的方法是: 通过 Qt Designer 放一个 第三方控件 的 上层layout, 然后在代码中 创建 第三方控件对象, 并且添加到 上层layout 里面去。
这在我们 实战班的 项目实战1 - 实战10
中有相应的锻炼。
当然也可以直接在 Qt Designer里面 放入第三方控件 ,这种方法其实更麻烦一些,如果你一定要这样做,可以点击这里,边看视频讲解,边学习如下内容
讲解视频,产生的界面ui文件在下面的链接zip文件中
https://www.byhy.net/cdn2/files/qt/stock-01.zip
如果你使用的是PySide2, 对应的代码如下,注意第14行 注册的作用
如果使用 PyQt5,就更简单了, 无需注册,对应的代码如下
from PyQt5.QtWidgets import QApplication
from PyQt5 import QtWidgets, uic
class Stock:
def __init__(self):
# PyQt5 直接加载ui文件
# 因为 第三方控件通过promote的定义已经知道控件类所在模块的路径
self.ui = uic.loadUi("main.ui")
hour = [1,2,3,4,5,6,7,8,9,10]
temperature = [30,32,34,32,33,31,29,32,35,45]
self.ui.historyPlot.plot(hour, temperature)
app = QApplication([])
stock = Stock()
stock.ui.show()
app.exec_()
轴刻度为字符串
上面的程序运行起来, X轴的刻度是 数字, 如果我们希望轴刻度是文字怎么做呢?
我们参考了这个网址的介绍: https://stackoverflow.com/questions/31775468/show-string-values-on-x-axis-in-pyqtgraph?lq=1
需要定义从数字到字符串的映射列表,参考如下代码
import pyqtgraph as pg
# 刻度表,注意是双层列表
xTick = [[(0, 'a'), (1, 'b'), (2, 'c'), (3, 'd'), (4, 'e'), (5, 'f')]]
x = [0, 1, 2, 3, 4, 5]
y = [1, 2, 3, 4, 5, 6]
win = pg.GraphicsWindow()
stringaxis = pg.AxisItem(orientation='bottom')
stringaxis.setTicks(xTick)
plot = win.addPlot(axisItems={'bottom': stringaxis})
curve = plot.plot(x,y)
pg.QtGui.QApplication.exec_()
如果使用 PlotWidget,则要获取轴对象,参考代码如下
# self.ui.historyPlot 就是 PlotWidget对象
xax = self.ui.historyPlot.getAxis('bottom')
xax.setTicks(xTick)
获取鼠标所在处刻度值
有时候,我们的程序需要获取 鼠标在 pyqtgraph 图形上移动时,鼠标所在对应的数据是什么。
解决方法,仅供实战班学员参考
练习题
开发一个股票数据软件,题目要求见 这个视频说明
VIP实战班学员联系老师获取参考代码。