Python数据可视化案例二：动态更新数据

在开发与数据监测和数据可视化有关的系统时，我们会需要根据最新的数据对图形进行更新。下面的代码模拟了这种情况，单击Start按钮时会更新数据并重新绘制图形使得曲线看上去在移动一样，单击Stop按钮则停止更新数据。

from time import sleep

from threading import Thread

import numpy as np

import matplotlib.pyplot as plt

from matplotlib.widgets import Button

fig, ax = plt.subplots()

#设置图形显示位置

plt.subplots_adjust(bottom=0.2)

#初始实验数据

range_start, range_end, range_step = 0, 1, 0.005

t = np.arange(range_start, range_end, range_step)

s = np.sin(4*np.pi*t)

l, = plt.plot(t, s, lw=2)

#自定义类，用来封装两个按钮的单击事件处理函数

class ButtonHandler:

def __init__(self):

self.flag = True

self.range_s, self.range_e, self.range_step = 0, 1, 0.005

#线程函数，用来更新数据并重新绘制图形

def threadStart(self):

while self.flag:

sleep(0.02)

self.range_s += self.range_step

self.range_e += self.range_step

t = np.arange(self.range_s, self.range_e, self.range_step)

ydata = np.sin(4*np.pi*t)

#更新数据

l.set_xdata(t-t[0])

l.set_ydata(ydata)

#重新绘制图形

plt.draw()

def Start(self, event):

self.flag = True

#创建并启动新线程

t = Thread(target=self.threadStart)

t.start()

def Stop(self, event):

self.flag = False

callback = ButtonHandler()

#创建按钮并设置单击事件处理函数

axprev = plt.axes([0.81, 0.05, 0.1, 0.075])

bprev = Button(axprev, 'Stop')

bprev.on_clicked(callback.Stop)

axnext = plt.axes([0.7, 0.05, 0.1, 0.075])

bnext = Button(axnext, 'Start')

bnext.on_clicked(callback.Start)

plt.show()