python np fft_用python学量子力学（2）

势垒贯穿

这一节研究一下量子力学里的势垒贯穿。为了完整运行本节 python 程序，需要将本文最后封装的 QuantumTunnel 类 copy 到 jupyter notebook 的最前面。

首先使用 np.heaviside 函数定义几个常见的势垒。heaviside(x, 0.5) 有两个参数，第一个参数 x 是一个数组，函数对 x < 0 区域返回 0，x>0 的区域返回 1，x==0 返回 0.5。

# 三种势垒的定义

使用上面这段代码，可以定义3种不同形状的势垒。先看下面这种阶梯状势垒，

%

初始波包的位置在 x=20 附近。波函数和概率密度如下图所示，

qt

调用类 QuantumTunnel 对象 qt 的 evolve() 函数，返回波包传播历史。

t

制作动画，演示波包从左边进入，部分穿透势垒，部分被反射的过程。代码如下，

%

知乎视频​www.zhihu.com

方势垒

%matplotlib inline
pot = lambda x: barrier(x, shape='square')

qt = QuantumTunnel(potential_wall = pot)

qt.plot_potential()

知乎视频​www.zhihu.com

从动画上看，进入势垒内部的粒子很难再穿透到右边真空区域，这与阶梯状势垒差别较大。如下图所示，

对于势阱，大部分穿过，小概率反射，展现量子力学的奇特行为。如下所示

完整动画

知乎视频​www.zhihu.com

改变阶梯状势垒的高度 U0 看反射概率

数值计算中，反射概率等于势垒左边的概率密度累加。

%

封装 QuantumTunnel 类

import 

总结：

封装了势垒贯穿类，测试在各种不同的势垒下，波包的穿透和反射过程。