TDSE Solution - Time Evaluation of WavePacket

Program 1

Time Evaluation of WavePacket

Python 3.11 Ready

'''Time evolution of Wavepacket'''
import numpy as np
import matplotlib.pyplot as plt
import matplotlib.animation as animation
from scipy.integrate import trapezoid

# Parameters
L, x0, k, a, N = 40.0, 4.0, 5.0, 2.0, 120
xs, h = np.linspace(0, L, N, retstep=True)
us = np.exp(-(xs-x0)**2/a) * np.exp(1.j*k*xs)
Vs = xs[1:-1]*0.0
dt = 0.1
mu = (0. + 1.j)*dt/(2*h**2)
t, tf = 0, 2.0

# Matrix for evolution
A1 = np.eye(N-2)*(1.0 + 2.0*mu + 1.j*dt/2*Vs) - mu*np.eye(N-2, k=1) - mu*np.eye(N-2, k=-1)
A2 = np.eye(N-2)*(1.0 - 2.0*mu + 1.j*dt/2*Vs) + mu*np.eye(N-2, k=1) + mu*np.eye(N-2, k=-1)
B = np.dot(np.linalg.inv(A1), A2)

for frame in range(int(tf/dt)):
    us[1:-1] = np.dot(B, us[1:-1])
    t += dt
    normus = us / np.sqrt(trapezoid(us * np.conj(us), xs))
    plt.clf()  
    plt.xlim(xs[0], xs[-1])
    plt.ylim(-0.6, 0.6)
    # Plot real part
    plt.plot(xs, np.real(normus), label='Real part')
    # Plot probability density
    plt.plot(xs, abs(normus)**2, label='|psi|Â²')
    plt.legend()
    plt.pause(0.01)

Click "Run Code" to execute script and render figures.

Mathematical Problem Formulation

Consider a one-dimensional quantum system described by the time-dependent SchrÃ¶dinger equation.

\[ i\hbar \frac{\partial \psi(x,t)}{\partial t} = \left( -\frac{\hbar^2}{2m} \frac{\partial^2}{\partial x^2} + V(x) \right) \psi(x,t) \]

Time evolution of a wave packet moving in free space. A Gaussian wave packet is defined as:

$$\psi(x, t=0) = A \cdot e^{-\frac{(x - x_0)^2}{2 \sigma^2}} \cdot e^{i k_0 x}$$

The energy of the wave packet is defined as:

$$E = \frac{\hbar^2 k_0^2}{2m} + \frac{\hbar^2}{2m \sigma^2}$$

Analyze Time evolution of the wave packet by Crank-Nicolson Method

Ch. 19.4 Time Evaluation of WavePacket

Time Evaluation of WavePacket

Mathematical Problem Formulation