Class for interactive plots

interactive_plot.py

import matplotlib.pyplot as plt

class InteractivePlot(object):

    def __init__(self, id, xlabel, ylabel):
        self.id = id
        plt.figure(id)
        plt.ion()
        self.xlabel = xlabel
        self.ylabel = ylabel
        self.x = []
        self.y = []

    def update(self, xval, yval):
        x = self.x
        y = self.y
        x.append(xval)
        y.append(yval)
        plt.figure(self.id)
        plt.clf()
        plt.plot(x, y, 'k')
        plt.xlabel(self.xlabel)
        plt.ylabel(self.ylabel)
        plt.draw()