我想开发一个具有附加吸引力的传播二维波函数的视觉模拟。基本上，我必须数值求解瞬态薛定谔方程（以 x、y、t 作为变量的 PDE），但是，我希望选择一个输入波函数作为初始波函数轮廓，而不是边界条件。什么软件会给我这样一个项目的良好环境（我认为 Mathematica、Comsol、Matlab 作为起点）？