misc VS schrod

Check out this interactive one! The Falstad applets don't get enough love. I also have old similar source here (Reddit thread).

I have a 1D schrodinger equation solver available on github. I use a conceptually identical approach here, but haven't yet packaged it to share. I will one day.

I have some source code for the 1D case on github.. That example uses the matplotlib library. For the 2D case, I'm actually using OpenCV instead. I followed this tutorial for writing a video file with opencv. In that tutorial, they read frames as numpy arrays off a video capture source (eg webcam), do something to it (eg grayscale the image), then write it to a file. In my case, I compute numpy arrays rather than read them off a capture source. Otherwise it's the same.

There are some extra complications from doing this in 2D rather than 1D, but the approach is conceptually identical. My code for the 1D case is open source and lives here: https://github.com/dhudsmith/schrod. There's also some more discussion on this on r/quantum: https://www.reddit.com/r/quantum/comments/lw9z9t/probability_distributions_for_the_first_36/.

I'll go halfway. The code used here is the direct 2D extension of the 1D solver I have on github here: https://github.com/dhudsmith/schrod. To extend to 2D, you just need a 2D basis for computing the Hamiltonian matrix elements. I use the 2D box basis as shared by u/ketarax above. The computation time goes up quite a bit.