Old high school files. Lessson notes/codes/projects etc.
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
|
|
|
|
import numpy as np
|
|
|
|
|
from scipy.constants import Planck, electron_mass
|
|
|
|
|
|
|
|
|
|
# constants
|
|
|
|
|
BOX_LENGTH = 0.01 # box length (meters)
|
|
|
|
|
h = Planck # plancks constant
|
|
|
|
|
|
|
|
|
|
D = np.sqrt(2/BOX_LENGTH) # norm factor
|
|
|
|
|
m = electron_mass # mass
|
|
|
|
|
|
|
|
|
|
# total energy function
|
|
|
|
|
def energy(n):
|
|
|
|
|
return (((h*n)/BOX_LENGTH)**2)*(1/(8*m))
|
|
|
|
|
|
|
|
|
|
# Wave func inner coef
|
|
|
|
|
def psi_k(n):
|
|
|
|
|
return np.sqrt( (8 * ((np.pi)**2) * m * energy(n))/(h**2) )
|
|
|
|
|
|
|
|
|
|
# pre calculate energy states
|
|
|
|
|
energy_def = []
|
|
|
|
|
for n in range(0, 11):
|
|
|
|
|
energy_def.append( energy(n) )
|
|
|
|
|
|
|
|
|
|
psi_k_def = []
|
|
|
|
|
for n in range(0, 11):
|
|
|
|
|
psi_k_def.append( psi_k(n) )
|
