add hw9 and hw8

1 files changed, 33 insertions, 6 deletions

diff --git a/hw8/10-3.jl b/hw8/10-3.jl
index 8616a69..2f4a690 100644
--- a/hw8/10-3.jl
+++ b/hw8/10-3.jl
@@ -60,16 +60,43 @@ function map_n_to_energies(n)
 	return e
 end
 
-n_max = 18
-n_to_e = [map_n_to_energies(n) for n in 1:n_max]
+n_s = collect(0:1:18)
+n_to_e = [map_n_to_energies(n) for n in n_s]
 
 # plot e[0] for all N
-eList = zeros(0)
-for i in 1:n_max
-	push!(eList, n_to_e[i][1])
+ground_state = []
+excited_1 = []
+excited_2 = []
+excited_3 = []
+for i in 1:length(n_to_e)
+	push!(ground_state, n_to_e[i][1])
+	push!(excited_1, n_to_e[i][2])
+	push!(excited_2, n_to_e[i][3])
+	push!(excited_3, n_to_e[i][4])
 end
 
-plot(eList)
+plot(ground_state, label = "groud state energy for n", xlabel = "n (level)", ylabel = "energy", title = "excited energy levels for V(n) = abs(x)^n", marker = :circle)
+plot!(excited_1, label = "1st excited state", marker = :circle)
+plot!(excited_2, label = "2nd excited state", marker = :circle)
+plot!(excited_3, label = "3rd excited state", marker = :circle)
+
+# plot the energies for an inifinite square well as a horizontial line
+# function excited_state_to_energy_inf_square_well(n)
+# 	return n^2 * pi^2 / 2
+# end
+
+# ground_state_inf_square_well = [excited_state_to_energy_inf_square_well(1) for i in 1:length(n_s)]
+# excited_1_inf_square_well = [excited_state_to_energy_inf_square_well(2) for i in 1:length(n_s)]
+# excited_2_inf_square_well = [excited_state_to_energy_inf_square_well(3) for i in 1:length(n_s)]
+# excited_3_inf_square_well = [excited_state_to_energy_inf_square_well(4) for i in 1:length(n_s)]
+
+# plot!(ground_state_inf_square_well, label = "ground state energy for infinite square well")
+# plot!(excited_1_inf_square_well, label = "1st excited state for infinite square well")
+# plot!(excited_2_inf_square_well, label = "2nd excited state for infinite square well")
+# plot!(excited_3_inf_square_well, label = "3rd excited state for infinite square well")
+
+
+savefig("hw8/10-3.png")
 
 # gs(x) = exp(-0.5 * x^2) # Gaussian that is exact ground state of SHO