In [2]:
import numpy as np
import matplotlib.pyplot as plt
import h5py
from fileloader import *
%matplotlib qt5
In [7]:
floc = "/home/zaph0d/git/Ahti/ahti-debug/bin/time_evo_0.h5"
psi = load_timestep(floc, 0)
dt = 0.001
Nt = 10000
E = 0.5
psi *= np.exp(-1j * E * dt * Nt)
floc = "../../ahti-debug/bin/time_evo_1.h5"
psi_t = load_timestep(floc, 1)
x = np.linspace(-6.0,6.0,1000)
plt.plot(x, np.abs(psi.imag-psi_t.imag))
plt.plot(x, np.abs(psi.real-psi_t.real))
plt.xlabel(r"Location x")
plt.ylabel(r"$\Psi$")
plt.show()
In [ ]: