Function probability_density_2d 

pub fn probability_density_2d(psi: &[Vec<Complex>]) -> Vec<Vec<f64>>

Probability density |psi|^2 for 2D wave function.