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()

#設(shè)置圖形顯示位置

plt.subplots_adjust(bottom=0.2)

#初始實(shí)驗(yàn)數(shù)據(jù)

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)

#自定義類，用來(lái)封裝兩個(gè)按鈕的單擊事件處理函數(shù)

class ButtonHandler:

    def __init__(self):

        self.flag = True

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

    #線程函數(shù)，用來(lái)更新數(shù)據(jù)并重新繪制圖形

    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)

            #更新數(shù)據(jù)

            l.set_xdata(t-t[0])

            l.set_ydata(ydata)

            #重新繪制圖形

            plt.draw()

    def Start(self, event):

        self.flag = True

        #創(chuàng)建并啟動(dòng)新線程

        t = Thread(target=self.threadStart)

        t.start()

    def Stop(self, event):

        self.flag = False

callback = ButtonHandler()

#創(chuàng)建按鈕并設(shè)置單擊事件處理函數(shù)

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()