Template for video from mpl plots

vid_from_mpl_plots.py

import matplotlib.pyplot as plt
import numpy as np
import imageio as iio

def my_plt_plotting_func(data):
  plt.plot(data)  # can be as extensive as you want, as long as it just makes a single pyplot fig

def write_frame(data, vid):
    my_plt_plotting_func(data)
    plt.savefig('current_frame.tiff')
    plt.close()
    t = iio.imread('current_frame.tiff')
    vid.append_data(t)
    
try:
    min_time = 0  # seconds
    max_time = 10
    step = 0.1   # seconds per frame
    width = 5  # how long to plot on x-axis
    fname = 'sniff_test.mp4'
    vid = iio.get_writer(fname, format='FFMPEG', mode='I', fps=1/step)
    for timestamp in np.arange(min_time, max_time, step):
        data = get_window_of_my_data(timestamp, timestamp+width)
        write_frame(data, vid)
    vid.close()
except KeyboardInterrupt:
    vid.close()