Would be nice if I knew I could implement this without double checking the correct syntax for matplotlib. Alas, my brain uses Mathematica.