示波器

模拟示波器。

示波器示例

  1. import numpy as np
  2. from matplotlib.lines import Line2D
  3. import matplotlib.pyplot as plt
  4. import matplotlib.animation as animation
  7. class Scope(object):
  8.     def __init__(self, ax, maxt=2, dt=0.02):
  9.         self.ax = ax
  10.         self.dt = dt
  11.         self.maxt = maxt
  12.         self.tdata = [0]
  13.         self.ydata = [0]
  14.         self.line = Line2D(self.tdata, self.ydata)
  15.         self.ax.add_line(self.line)
  16.         self.ax.set_ylim(-.1, 1.1)
  17.         self.ax.set_xlim(0, self.maxt)
  19.     def update(self, y):
  20.         lastt = self.tdata[-1]
  21.         if lastt > self.tdata[0] + self.maxt:  # reset the arrays
  22.             self.tdata = [self.tdata[-1]]
  23.             self.ydata = [self.ydata[-1]]
  24.             self.ax.set_xlim(self.tdata[0], self.tdata[0] + self.maxt)
  25.             self.ax.figure.canvas.draw()
  27.         t = self.tdata[-1] + self.dt
  28.         self.tdata.append(t)
  29.         self.ydata.append(y)
  30.         self.line.set_data(self.tdata, self.ydata)
  31.         return self.line,
  34. def emitter(p=0.03):
  35.     'return a random value with probability p, else 0'
  36.     while True:
  37.         v = np.random.rand(1)
  38.         if v > p:
  39.             yield 0.
  40.         else:
  41.             yield np.random.rand(1)
  43. # Fixing random state for reproducibility
  44. np.random.seed(19680801)
  47. fig, ax = plt.subplots()
  48. scope = Scope(ax)
  50. # pass a generator in "emitter" to produce data for the update func
  51. ani = animation.FuncAnimation(fig, scope.update, emitter, interval=10,
  52.                               blit=True)
  54. plt.show()

下载这个示例