9个动图带你进入PyQtGraph的强大可视化世界

PyQtGraph是一个建立在PyQt/PySide之上的Python数据可视化图形界面库,其性能强、速度快,能够胜任大部分交互式的2D、3D图形绘制,可以搞定数据科学领域大量的数据可视化工作。
州的先生之前绘制的A股股票动态K线图,就是使用PyQtGraph绘制的。
A股复苏,入场还是解套?Python秒绘交互式股票K线图
PyQtGraph官方提供了很好的示例来供学习者了解PyQtGraph的功能,下面我们通过9个图形,来一窥PyQtGraph的可视化世界。

基础的数组绘制折线图

# coding:utf-8 # 作者:州的先生 # 博客:https://zmister.com from pyqtgraph.Qt import QtGui, QtCore import numpy as np import pyqtgraph as pg # 实例化一个绘图窗口 win = pg.GraphicsWindow() win.resize(1000,600) win.setWindowTitle('PyQtGraph基础绘图示例 - zmister.com') # 启用抗锯齿选项 pg.setConfigOptions(antialias=True) # 使用addPlot()方法直接添加一个数组以绘制图形 p1 = win.addPlot(title="基础数组绘图", y=np.random.normal(size=100)) if __name__ == '__main__':     import sys     if (sys.flags.interactive != 1) or not hasattr(QtCore, 'PYQT_VERSION'):         QtGui.QApplication.instance().exec_()

运行上述代码,会得到如下动图所示的图形界面:

一个图形中绘制多条折线

# coding:utf-8 # 作者:州的先生 # 博客:https://zmister.com from pyqtgraph.Qt import QtGui, QtCore import numpy as np import pyqtgraph as pg # 实例化一个绘图窗口 win = pg.GraphicsWindow() win.resize(1000,600) win.setWindowTitle('PyQtGraph基础绘图示例 - zmister.com') # 启用抗锯齿选项 pg.setConfigOptions(antialias=True) # 添加一个图形 p2 = win.addPlot(title="多条折线") # 在图形中绘制3条折线 p2.plot(np.random.normal(size=100), pen=(255,0,0), name="红色线条") p2.plot(np.random.normal(size=110)+5, pen=(0,255,0), name="绿色线条") p2.plot(np.random.normal(size=120)+10, pen=(0,0,255), name="蓝色线条") if __name__ == '__main__':     import sys     if (sys.flags.interactive != 1) or not hasattr(QtCore, 'PYQT_VERSION'):         QtGui.QApplication.instance().exec_()

运行上述代码,会得到如下动图所示的图形界面:

在图形中绘制点

# coding:utf-8 # 作者:州的先生 # 博客:https://zmister.com from pyqtgraph.Qt import QtGui, QtCore import numpy as np import pyqtgraph as pg # 实例化一个绘图窗口 win = pg.GraphicsWindow() win.resize(1000,600) win.setWindowTitle('PyQtGraph基础绘图示例 - zmister.com') # 启用抗锯齿选项 pg.setConfigOptions(antialias=True) # 添加一个图形 p3 = win.addPlot(title="在图形中绘制点") p3.plot(     np.random.normal(size=100), # 随机数组曲线     pen=(200,200,200), # 画笔颜色     symbolBrush=(0,255,0),  # 符号画笔颜色     symbolPen='w', # 符号外圈颜色 ) if __name__ == '__main__':     import sys     if (sys.flags.interactive != 1) or not hasattr(QtCore, 'PYQT_VERSION'):         QtGui.QApplication.instance().exec_()

运行上述代码,会得到如下动图所示的图形界面:

使用图形参数

# coding:utf-8 # 作者:州的先生 # 博客:https://zmister.com from pyqtgraph.Qt import QtGui, QtCore import numpy as np import pyqtgraph as pg # 实例化一个绘图窗口 win = pg.GraphicsWindow() win.resize(1000,600) win.setWindowTitle('PyQtGraph基础绘图示例 - zmister.com') # 启用抗锯齿选项 pg.setConfigOptions(antialias=True) # 添加一个图形 p4 = win.addPlot(title="使用参数来显示网格") x = np.cos(np.linspace(0, 2*np.pi, 1000)) # 生成X轴数据 y = np.sin(np.linspace(0, 4*np.pi, 1000)) # 生成Y轴数据 p4.plot(x, y) # 绘制图形 p4.showGrid(x=True, y=True) # 启用X、Y轴网格 if __name__ == '__main__':     import sys     if (sys.flags.interactive != 1) or not hasattr(QtCore, 'PYQT_VERSION'):         QtGui.QApplication.instance().exec_()

运行上述代码,会得到如下动图所示的图形界面:

散点图、坐标轴标签和刻度

# coding:utf-8 # 作者:州的先生 # 博客:https://zmister.com from pyqtgraph.Qt import QtGui, QtCore import numpy as np import pyqtgraph as pg # 实例化一个绘图窗口 win = pg.GraphicsWindow() win.resize(1000,600) win.setWindowTitle('PyQtGraph基础绘图示例 - zmister.com') # 启用抗锯齿选项 pg.setConfigOptions(antialias=True) # 添加一个图形 p5 = win.addPlot(title="散点图、坐标轴标签、坐标轴刻度") x = np.random.normal(size=1000) * 1e-5 # 生成X轴数据 y = x*1000 + 0.005 * np.random.normal(size=1000) # 生成Y轴数据 y -= y.min()-1.0 mask = x > 1e-15 x = x[mask] y = y[mask] p5.plot(x, y, pen=None, symbol='t', symbolPen=None, symbolSize=10, symbolBrush=(100, 100, 255, 50)) p5.setLabel('left', "Y Axis", units='A') # 设置横坐标轴标签文本 p5.setLabel('bottom', "Y Axis", units='s') #设置纵坐标轴标签文本 p5.setLogMode(x=True, y=False) # 设置坐标轴刻度模式 if __name__ == '__main__':     import sys     if (sys.flags.interactive != 1) or not hasattr(QtCore, 'PYQT_VERSION'):         QtGui.QApplication.instance().exec_()

运行上述代码,会得到如下动图所示的图形界面:

实时数据更新绘图

# coding:utf-8 # 作者:州的先生 # 博客:https://zmister.com from pyqtgraph.Qt import QtGui, QtCore import numpy as np import pyqtgraph as pg # 实例化一个绘图窗口 win = pg.GraphicsWindow() win.resize(1000,600) win.setWindowTitle('PyQtGraph基础绘图示例 - zmister.com') # 启用抗锯齿选项 pg.setConfigOptions(antialias=True) # 添加一个图形 p6 = win.addPlot(title="绘图数据更新") curve = p6.plot(pen='y') # 图形使用黄色画笔进行绘制 data = np.random.normal(size=(10,1000)) # 生成随机数据 ptr = 0 # 初始为0 # 定义一个更新函数 def update():     global curve, data, ptr, p6     curve.setData(data[ptr%10]) # 设置图形的数据值     if ptr == 0:         p6.enableAutoRange('xy', False)  ## 在第一个图形绘制的时候停止自动缩放     ptr += 1 timer = QtCore.QTimer() # 实例化一个计时器 timer.timeout.connect(update) # 计时器信号连接到update()函数 timer.start(200) # 计时器间隔200毫秒 if __name__ == '__main__':     import sys     if (sys.flags.interactive != 1) or not hasattr(QtCore, 'PYQT_VERSION'):         QtGui.QApplication.instance().exec_()

运行上述代码,会得到如下动图所示的图形界面:

填充图形、禁用坐标轴

# coding:utf-8 # 作者:州的先生 # 博客:https://zmister.com from pyqtgraph.Qt import QtGui, QtCore import numpy as np import pyqtgraph as pg # 实例化一个绘图窗口 win = pg.GraphicsWindow() win.resize(1000,600) win.setWindowTitle('PyQtGraph基础绘图示例 - zmister.com') # 启用抗锯齿选项 pg.setConfigOptions(antialias=True) # 添加一个图形 p7 = win.addPlot(title="填充图形,禁用坐标轴") y = np.sin(np.linspace(0, 10, 1000)) + np.random.normal(size=1000, scale=0.1) # 生成图形数据 p7.plot(y,     fillLevel=0,     brush=(50,50,200,100) ) p7.showAxis('bottom', False) # 底部坐标轴不显示 if __name__ == '__main__':     import sys     if (sys.flags.interactive != 1) or not hasattr(QtCore, 'PYQT_VERSION'):         QtGui.QApplication.instance().exec_()

运行上述代码,会得到如下动图所示的图形界面:

区域选择

# coding:utf-8 # 作者:州的先生 # 博客:https://zmister.com from pyqtgraph.Qt import QtGui, QtCore import numpy as np import pyqtgraph as pg # 实例化一个绘图窗口 win = pg.GraphicsWindow() win.resize(1000,600) win.setWindowTitle('PyQtGraph基础绘图示例 - zmister.com') # 启用抗锯齿选项 pg.setConfigOptions(antialias=True) # 添加一个图形 x2 = np.linspace(-100, 100, 1000) data2 = np.sin(x2) / x2 p8 = win.addPlot(title="区域选择") p8.plot(data2, pen=(255,255,255,200)) # 添加一个线区域选择项目,起始区间在400到700 lr = pg.LinearRegionItem([400,700]) p8.addItem(lr) if __name__ == '__main__':     import sys     if (sys.flags.interactive != 1) or not hasattr(QtCore, 'PYQT_VERSION'):         QtGui.QApplication.instance().exec_()

运行上述代码,会得到如下动图所示的图形界面:

放大区域选择

# coding:utf-8 # 作者:州的先生 # 博客:https://zmister.com from pyqtgraph.Qt import QtGui, QtCore import numpy as np import pyqtgraph as pg # 实例化一个绘图窗口 win = pg.GraphicsWindow() win.resize(1000,600) win.setWindowTitle('PyQtGraph基础绘图示例 - zmister.com') # 启用抗锯齿选项 pg.setConfigOptions(antialias=True) # 添加一个图形 x2 = np.linspace(-100, 100, 1000) data2 = np.sin(x2) / x2 p8 = win.addPlot(title="区域选择") p8.plot(data2, pen=(255,255,255,200)) # 添加一个线区域选择项目,起始区间在400到700 lr = pg.LinearRegionItem([400,700]) p8.addItem(lr) p9 = win.addPlot(title="放大区域选择") p9.plot(data2) # 更新绘图 def updatePlot():     p9.setXRange(*lr.getRegion(), padding=0) # 更新区域选择 def updateRegion():     lr.setRegion(p9.getViewBox().viewRange()[0]) lr.sigRegionChanged.connect(updatePlot) p9.sigXRangeChanged.connect(updateRegion) updatePlot() if __name__ == '__main__':     import sys     if (sys.flags.interactive != 1) or not hasattr(QtCore, 'PYQT_VERSION'):         QtGui.QApplication.instance().exec_()

运行上述代码,会得到如下动图所示的图形界面:

全家福

最后,来一个全家福:

分类目录: 编程

标签: pyqtgraph