alkesh

quantum1.py

import numpy as np
import matplotlib.pyplot as plt
z=1j
x=np.linspace(-30,30,1000) # 1000 equidistant points in [-30,30]
x0=19 #position of the barrier

psi0=np.sin(x0)


psii=np.zeros(np.size(x)) #for incident
psir=np.zeros(np.size(x)) # for refelcted


for pos in range(np.size(x)):
    x_val=x[pos]
    if (x_val<=x0): # before barrier
        psii[pos]=np.sin(x_val)
        psir[pos]=psi0*np.cos(x_val-x0)
    else: # after barrier
        psit=psi0*np.exp(-(x_val-x0)*z).real #taking the real part
        psii[pos]=psit
        psir[pos]=psit

        
        
plt.figure(figsize=[12,8])
plt.axvline(x=x0,color="black",linestyle="--")
plt.grid()
plt.ylim((-1.5,1.5))
plt.xlabel("x (nm)")
plt.ylabel("psi ")


plt.text(x0,1,"boundary",size=15)
plt.plot(x,psii,label="incident")
plt.plot(x,psir,label="reflected")
plt.legend()