我正在处理一个代码，其中涉及使用 Crank-Nicolson 时间步求解一维薛定谔方程。代码是用 NumPy/SciPy 编写的，我做了一些分析，发现大部分模拟时间都被调用线性求解器（在本例中为 spsolve）占用了。我正在使用简单的离散化，这样我的矩阵是对称的和三对角的，我想知道是否有人尝试过直接实现 Thomas 算法之类的事情，以及这是否能够击败 spsolve。