华中师范大学 hahakity
在网上看到一个使用 Matlab 教量子力学的文章,很有意思。这里用 python 语言实现一遍, 让同学们对量子力学,对偏微分方程的差分近似解法有一个更直观的理解。
学习目标:理解量子力学的波函数表示与矩阵表示的等价性
学会用向量表示函数,用矩阵表示算符(一阶微分,二阶微分)
学会数值求解任意势阱下定态薛定谔方程的能级与波函数
预备知识:微分的差分近似
量子力学基础(薛定谔方程)
波函数的向量表示
在用 python 画图时,我们一般先将区间离散化,计算出离散坐标上的函数值,然后画折线图。比如对于函数
, 使用如下代码画图,
# np.linspace 将区间 [-2, 2] 离散化为 100 个坐标点
x = np.linspace(-2, 2, 100)
# 计算 100 个坐标点上的函数值 f
f = np.sin(x) / x
plt.plot(x, f)
将波函数表示为离散坐标点上的实数或复数,写为列向量
变得非常容易理解。
回忆微分的有限差分近似,对于一阶微分,
对于二阶微分,
对区间 [a, b] 所有离散坐标上的 f(x) 微分和二阶微分可以矩阵化,
算符的矩阵表示
当 f(x) 用列向量
表示时,就可以用矩阵来表示微分算子。
因此波动力学与矩阵力学统一。
一阶微分
可以用微分算子矩阵 D 点乘
计算。
二阶微分
可以用微分算子矩阵 D 从左边连续作用两次到
上,也可以使用差分格式直接构造拉普拉斯矩阵,由
计算。
对于随便给定的函数
, 可以看到上述有限差分矩阵作用在离散的
上的结果与解析解非常一致。
解定态薛定谔方程
的任务转化为求哈密顿矩阵 H 的本征值 E 和本征向量
。
注意在量子力学里,动能项里的动量 p 换成了微分算符,进一步用 Laplacian 矩阵表示,势能 U(x) 在区间离散化后,填充在矩阵的对角元上。
如果是多粒子系统,比如考虑多个电子两两之间的库伦相互作用,则两粒子势能
会带来非对角元。
定态薛定谔方程的数值解
这里我用 python 把一维定