Create a matplotlib FuncAnimation from a function keeps throwing shape mismatch

Question

I\'m trying to use FuncAnimation to plot a wavefunction, and I have my function working fine in a standard plt.plot(x,y) situation:

import numpy as n