Laboratorio 2 - Soluciones numéricas a la ecuación de

Schrodinger
Mecánica Cuántica I - CF3F2 A

Alexander A. Trigoso Sulca, Jose G. Huacachi Bazán, Carlos D. Palante Palomino

November 1, 2022

Resumen
Se usó el programa Sagemath para la resolución de problemas relacionados a la mecánica
cuántica. Se utilizó métodos numéricos para la soución de la ecuación de Shrodinger. De este
modo, se determinó las funciones propias y sus energı́as propias de manera numérica.

Abstract
The Sagemath program was used to solve problems related to quantum mechanics. Numerical
methods were used to solve the Shrodinger equation. In this way, the eigenfunctions and their
eigenenergies were determined numerically.

1 Introducción
La resolución de problemas relacionados con la mecánica cuántica es nuestro tema principal usando
el programa Sagemath. Esto es importante debido a que a menudo la ciencia necesita a veces agilizar
cálculos extensos para poder seguir avanzando. El uso de este tipo de programas es extenso y muy
necesario en el amplio mundo de la fı́sica.

2 Marco teórico
2.1 Densidad de corriente o flujo de probabilidad
Se define como  
−ih̄ dψ ψ∗
jx = ψ∗ −ψ
2m dx dx

2.2 La aproximación WKB

La aproximación WKB (Wentzel, Kramers, Brillouin) es un método para obtener soluciones aproxi-
madas a la ecuación unidimensional de Schrodinger independiente del tiempo. Sea una partı́cula con
energı́a E que se mueve en una región donde el potencial es V (x) y constante. Si E > V entonces la
fucnión de onda es de la forma
ψ(x) = Ae±ikx
p
donde k = 2m(E − V )/h̄. La solución general es una combinación lineal de las soluciones con el
signo del exponente opuestos. La solución es oscilante con una longitud de onda λ = 2π/k y una
amplitud A que no cambia.

Supongamos ahora que el potencial V (x) no es constante, varı́a poco en comparación de λ. Entonces
sobre esa región el potencial se puede considerar constante. Es razonable suponer que ψ permanece
casi sinusoidal, excepto cuando la longitud de onda y la amplitud cambian pequeñamente con x.

1
 Para regiones clásicas, espdecir, donde E > V (x), entonces de la ecuación de Schrodinger podemos
hacer la sustitución p(x) = 2m[E − V (x)]. Quedandonos:

d2 ψ p2
2
= − 2ψ (1)
dx h̄
Por lo tanto p(x) es real. En general la solución de la función de onda será:

ψ(x) = A(x)eϕ(x) (2)

Reemplazando la ecuación X en la Y se obtiene

p2
A′′ + 2iA′ ϕ′ + iAϕ′′ − A(ϕ′ )2 = − A
h̄2
Separando la parte real e imaginaria se llega

p2
A′′ − A(ϕ′ )2 = −
A (3)
h̄2
 
′′ ′ 2 p2
A = A (ϕ ) − 2 (4)
h̄
2A′ ϕ′ − Aϕ′′ = 0 (5)
2 ′ ′
De la ecuación X’ se resuelve que (A ϕ ) = 0 y por esto asumiendo C una constante:

C
A= √ ′ (6)
ϕ

En la ecuación X” asumimos que la amplitud A varia pequeñamente por lo que despreciamos A′′ /A.
Entonces:
dϕ p
=± (7)
dx h̄
Por lo tanto Z
1
ϕ(x) = ± p(x)dx + C ′ (8)
h̄
Reemplazando en la función de onda tenemos
C i
R
(x) = p e± h̄ p(x)dx (9)
p(x)

La solución general es aquella que es una combinación lineal de esta solucion son los signos opuestos
del exponente. Notemos que
|C|2
|ψ(x)|2 =
p(x)
lo cual dice que la probabilidad de encontrar a la particula en el punto x es inversamente proporcional
a su momentum en dicho punto. Esto es exactamente lo que uno esperarı́a, la partı́cula no pasa mucho
tiempo en los lugares donde se mueve rápidamente, es por eso que la probabilidad es pequeña.

2.2.1 Potencial con dos paredes verticales

Supongamos que tenemos un pozo de potencial infinito con un potencial dentro como en la figura X.
Formalmente


 ∞ si x≤0



V (x) = Ep (x) si 0 < x < a





∞ si x≥a

2
 Figure 1: Pozo infinito de potencial con un potencial en pa parte inferior

Asumiendo que E > V (x) dentro de las paredes, entonces

1 h i
ψ=p C+ eiϕ(x) + C− e−iϕ(x)
p(x)

De forma conveniente podemos optar por la forma:

1
ψ=p [C1 sinϕ(x) + C2 cosϕ(x)]
p(x)

donde Z x
1
ϕ(x) = p(x′ )dx′
h̄ 0

Ahora por condiciones de frontera, em x = 0 la fucnion de onda ψ(0) = 0. Por lo tanto C2 = 0.

Además en x = a debe ser también ψ(a) = 0. Por lo tanto

ϕ(a) = nπ (n = 1, 2, 3, ..)

En conclusión: Z a
p(x)dx = nπh̄ (10)
0
Esta cuanticación como condición permite determinar aproximadamente las energı́as permitidas.

2.3 Barrera de potencial

Se tiene un potencial escalón especificado de la siguiente manera:

 0 si x ≤ 0
V (x) =

V0 si 0 < x
Se puede dar el caso donde la energı́a de las partı́culas que estan incidiendo desde la izquierda sobre
el potencial escalón es mayor o menor que el valor del potencial. Para cada una de estas situaciones
los coeficientes de transmisión y reflexión son distintos.
Para E < V0 se tiene:
1
ψ=p [C1 sinϕ(x) + C2 cosϕ(x)]
p(x)

3
 1 h i
ψ=p C+ eiϕ(x) + C− e−iϕ(x)
p(x)

 A+ eiϕ(x) si x≤0
ψ(x) =

V0 si 0 < x

R=1 (11)

2.4 Justificación
Usar programas como Sagemath para realizar cálculos simbólicos y numéricos agiliza la recolección de
patrones.

2.5 Antecedentes
El siglo pasado no se usaban computadoras para calcular operaciones matemáticas tan complejas como
las actuales. Estos cálculos se hacı́an a mano y podrı́a resultar muy tedioso y estar propenso a errores
humanos.

4
3 Metodologı́a y/o desarrollo experimental
Los problemas del laboratorio fueron tratados de manera manual previamente, posteriormente se utilizó
Sagemath ya sea para aligerar el proceso de calculo o comprobacion y verificacion.

5
4 Resultados y análisis

6
 Laboratorio 2 - Problema 1

October 31, 2022

[1]: %display latex

Demostrar que el flujo de partículas (corriente) es cero si la función de onda unidimensional de una
partícula es:
2
ψ(x, t) = be−ax −iωt
donde a y b son números reales positivos.
Simpleemnte por definición de flujo de probabilidad o densidad de corriente:
( )
−ih̄ ∗ dψ dψ ∗
jx = ψ −ψ
2m dx dx

[6]: hbar = var('hbar', latex_name=r'\mathcal{\hbar}')

m, a, b, t, w = var('m a b t w')

[18]: assume(a>0)
assume(b>0)
assume(w>0)
assume(x, 'real')
assume(t, 'real')

[19]: #definición de la función de onda

psi(x,t) = b*e^(-a*x^2-i*w*t)
psi(x,t)
[19]:
be(−ax )
2 −i tw

[22]: j(x)=-i*hbar/(2*m)*( psi(x,t).conjugate()*diff(psi(x,t),x) -␣

,→psi(x,t)*diff(psi(x,t).conjugate(),x) )

[23]: j(x)
[23]:
0

1
 Laboratorio 2 - Problema 3 - Alexander Trigoso Sulca

October 31, 2022

[1]: %display latex

Calcule numericamente dos valores propio de energia y sus respectivas funciones propias para un
electrón con masa efectiva me , confinado en un pozo de potencial de ancho L, delimitado por
barreras de energía infinita en x < 0 y x > L. EL pozo de potencial en la parte interna es de la
forma
V (x) = Vo x2 /L2
Explique el cambio de la forma de la función de onda con el aumento de la energía propia. Emplear
me = 1, Vo = 1, L = 20, h̄ = 1.
Se usará el método de aproximación WKB expuesto en el marco teórico para una región clásica, es
decir, para energías mayores al potencial.

[90]: # variables constantes

m, L = var('m L')
Vo = var('Vo', latex_name = r'\mathcal{V_o}')
hbar = var('hbar', latex_name = r'\mathcal{\hbar}')

[91]: # variables independientes

x = var('x')
# función potencial
V(x) = Vo*x^2/L^2

[92]: # Grafica de la función potencial

V_plot=plot(V(x).substitute(Vo = 1, L = 20), 0, 20, color = 'purple')
V_plot.show()

1
[93]: # definicion de la energia o valor propio
E = var('E')
assume(E, 'real')
assume(E>1) # mayor al potencial V(x)

[6]: # función p(x)

p(x) = sqrt(2*m*(E-V(x).substitute(Vo = 1, L = 20)))
p(x)
[6]:
√
1 1√
−(x2 − 400 E)m
10 2

[7]: # funcion phi(y)

y = var('y')
assume(y>0)
assume(y<20)
assume(x<20)
phi(y) = integrate( p(x).substitute(m=1), [x, 0, y]) # hbar=1
phi(y)
[7]:
( ( ) √ )
1 √ y 2
2 400 E arcsin √ + −y + 400 Ey
40 20 E

2
 [94]: # condicion de frontera de cuantización
cf = integrate(p(x).substitute(m=1), [x, 0, 20])
cf.show()

( ( ) )
√ 1 √
10 2 E arcsin √ + E−1
E

[101]: # f(E) = n*pi*hbar

plot(cf, [E, 1,2], color = 'purple')+plot(8*pi, 1, 2, color ='aqua',␣
,→legend_label='n=8')+plot(9*pi, 1, 2, color ='green',␣

,→legend_label='n=9')+plot(10*pi,1, 2, color ='blue',␣

,→legend_label='n=10')+plot(11*pi, 1,2, color ='black', legend_label='n=11')

[101]:

0.1 Valores propios

[102]: # Para n=8:
E8 = find_root(8*pi==cf,0,5)
E8
[102]:
1.156993776597359

3
[103]: # Para n=9:
E9 = find_root(9*pi==cf,0,5)
E9
[103]:
1.358045385217778

[104]: # Para n=10:

E10 = find_root(10*pi==cf,0,5)
E10
[104]:
1.5869459565084063

[105]: # Para n=11:

E11 = find_root(11*pi==cf,0,5)
E11
[105]:
1.8422166132807896

0.2 Función de onda de manera general

[15]: # la funcion de onda
psi = var('psi', latex_name=r'\mathcal{\psi}')
C = var('C')
psi(x) = C*(sin(phi(x)))/sqrt(p(x).substitute(m=1))
psi(x)
[15]:
(1)3 √ 1
(
1
√ ( ( ) √ ))
20 2
4
10 C sin 40 2 400 E arcsin 20 x√E + −x2 + 400 Ex
1
(−x2 + 400 E) 4

0.3 Para n =8
[109]: # Para n = 8 :
psi_8(x) = (sin(phi(x).substitute(E=E8)))/sqrt(p(x).substitute(m=1,E=E8))
psi_8(x)
[109]:

(1)3 √ 1
(
1
√ (√ ))
20 2
4
10 sin 40 2 −x2 + 462.7975106389436x + 462.7975106389436 arcsin (0.04648410707255946 x)
1
(−x2 + 462.7975106389436) 4

[116]: numerical_integral(psi_8(x)*psi_8(x).conjugate(), (0,20))

[116]:
( )
8.230880897852947, 9.138113485743722 × 10−14

4
 Escogiendo el C positivo:
C 2I = 1
√
C = 1/ I

[117]: # Entonces la funcion de onda par a n=8 es:

psi(x).substitute(C=1/sqrt(8.230880897852948), E = E8)
[117]:
(1)3 √ 1
(
1
√ (√
6.97118867323377 2
4
10 sin 40 2 −x2 + 462.7975106389436x + 462.7975106389436 arcsin (0.04648410707
1
(−x2 + 462.7975106389436) 4

[45]: psi_8_plot = plot(psi(x).substitute(C=1/sqrt(8.230880897852948), E = E8), 0,20)

psi_8_plot+V_plot+plot(E8, 0,20)
[45]:

0.4 Para n=9

[49]: psi_9(x) = (sin(phi(x).substitute(E=E9)))/sqrt(p(x).substitute(m=1,E=E9))
psi_9(x)
[49]:
(1)3 √ 1
(
1
√ (√ ))
20 2
4
10 sin 40 2 −x2 + 543.2181540871112x + 543.2181540871112 arcsin (0.04290548961595624 x)
1
(−x2 + 543.2181540871112) 4

5
[50]: numerical_integral(psi_9(x)*psi_9(x), (0,20))
[50]:
( )
7.246840611011792, 8.69292567671969 × 10−13

[118]: # Entonces la funcion de onda par a n=9 es:

psi(x).substitute(C=1/sqrt(7.246840611011792), E = E9)
[118]:
(1)3 √ 1
(
1
√ (√
7.42943249432094 2
4
10 sin 40 2 −x2 + 543.2181540871112x + 543.2181540871112 arcsin (0.04290548961
1
(−x2 + 543.2181540871112) 4

[51]: psi_9_plot = plot(psi(x).substitute(C=1/sqrt(7.246840611011792), E = E9), 0,20)

psi_9_plot+V_plot+plot(E9, 0,20)
[51]:

0.5 Para n =10

[52]: psi_10(x) = (sin(phi(x).substitute(E=E10)))/sqrt(p(x).substitute(m=1,E=E10))
psi_10(x)
[52]:
(1)3 √ 1
(
1
√ (√ ))
20 2
4
10 sin 40 2 −x2 + 634.7783826033625x + 634.7783826033625 arcsin (0.039690716215922586 x)
1
(−x2 + 634.7783826033625) 4

6
[53]: numerical_integral(psi_10(x)*psi_10(x), (0,20))
[53]:
( )
6.468849551034913, 5.716541756486749 × 10−11

[119]: # Entonces la funcion de onda par a n=10 es:

psi(x).substitute(C=1/sqrt(6.468849551034913), E = E10)
[119]:
(1)3 √ 1
(
1
√ (√
7.86351048728763 2
4
10 sin 40 2 −x2 + 634.7783826033625x + 634.7783826033625 arcsin (0.03969071621
1
(−x2 + 634.7783826033625) 4

[54]: psi_9_plot = plot(psi(x).substitute(C=1/sqrt(6.468849551034913), E = E10), 0,20)

psi_9_plot+V_plot+plot(E10, 0,20)
[54]:

0.6 Para n = 11
[55]: psi_11(x) = (sin(phi(x).substitute(E=E11)))/sqrt(p(x).substitute(m=1,E=E11))
psi_11(x)
[55]:
(1)3 √ 1
(
1
√ (√ ))
20 2
4
10 sin 40 2 −x2 + 736.8866453123159x + 736.8866453123159 arcsin (0.03683830651105816 x)
1
(−x2 + 736.8866453123159) 4

7
[56]: numerical_integral(psi_11(x)*psi_11(x), (0,20))
[56]:
( )
5.850173162712695, 2.250384047460604 × 10−08

[120]: # Entonces la funcion de onda par a n=12 es:

psi(x).substitute(C=1/sqrt(5.850173162712695), E = E11)
[120]:

(1)3 √ 1
(
1
√ (√
8.26885992580878 2
4
10 sin 40 2 −x2 + 736.8866453123159x + 736.8866453123159 arcsin (0.03683830651
1
(−x2 + 736.8866453123159) 4

[121]: psi_9_plot = plot(psi(x).substitute(C=1/sqrt(5.850173162712695), E = E11), 0,20)

psi_9_plot+V_plot+plot(E11, 0,20)
[121]:

[122]: PT8 = plot(psi(x).substitute(C=1/sqrt(8.230880897852948), E = E8), 0,20, color␣

,→= 'aqua', legend_label = 'Estado n=8')

PT9 = plot(psi(x).substitute(C=1/sqrt(7.246840611011792), E = E9), 0,20, color␣

,→= 'green', legend_label = 'Estado n=9')

PT10 = plot(psi(x).substitute(C=1/sqrt(6.468849551034913), E = E10), 0,20,␣

,→color = 'blue', legend_label = 'Estado n=10')

8
 PT11 = plot(psi(x).substitute(C=1/sqrt(5.850173162712695), E = E11), 0,20,␣
,→color = 'brown', legend_label = 'Estado n=11')

PT = PT8+PT9+PT10+PT11
PT.show()

[127]: PT8 = plot(psi(x).substitute(C=1/sqrt(8.230880897852948), E = E8), 0,20, color␣

,→= 'aqua', legend_label = 'Estado n=8')

PT11 = plot(psi(x).substitute(C=1/sqrt(5.850173162712695), E = E11), 0,20,␣

,→color = 'brown', legend_label = 'Estado n=11')

PT8+PT11+V_plot
[127]:

9
 Como podemos observar, cuando aumentamos el valor de n, es decir,la energía
propia, la función de onda cambia en su longitud de onda. Si aumentamos la
energía, la longitud de onda disminuye pero conforme avanzamos en x, esta tiende
a cercer. Para valores muy grandes de n, la energia será cuantizada de tal forma
que el potencial en la región interna se desprecie.

0.7 Para n=15

[123]: E15 = find_root(15*pi==cf,0,5)
E15
[123]:
3.1174769564134572

[124]: psi_15(x) = (sin(phi(x).substitute(E=E15)))/sqrt(p(x).substitute(m=1,E=E15))

psi_15(x)
[124]:

(1)3 √ 1
(
1
√ (√ ))
20 2
4
10 sin 40 2 −x2 + 1246.990782565383x + 1246.990782565383 arcsin (0.02831837824992808 x)
1
(−x2 + 1246.990782565383) 4

[125]: numerical_integral(psi_15(x)*psi_15(x), (0,20))

[125]:

10
 ( )
4.256755287858642, 4.726774527341604 × 10−14

[126]: psi_15_plot = plot(psi(x).substitute(C=1/sqrt(4.256755287858642), E = E15),␣

,→0,20)

psi_15_plot+V_plot+plot(E15, 0,20)
[126]:

11
 Laboratorio 2 - Problema 4 -Jose Huacachi

November 1, 2022

[4]: %display latex

[5]: #Definiendo variables

[6]: k,h,j,a,m_1,m_2 = var('k,h,j,a,m_1,m_2')

assume(k,'real')
assume(h,'real')
assume(j,'real')
assume(m_1,'real')
assume(m_2,'real')
assume(k>0,h>0,j>0,m_1>0,m_2>0)
Ener=var('Ener',latex_name=r'\mathcal{E}')
hbar=var('hbar',latex_name=r'\hbar')
V=var('V',latex_name=r'V')
psi = function('psi')(x)

#Ecuaciones a resolver

eqs_1 = diff(psi, x, 2) + k^(2)*psi == 0 #Para E>V

eqs_2 = diff(psi, x, 2) + h^(2)*psi == 0
eqs_3 = diff(psi, x, 2) - j^(2)*psi == 0 # E<V

show(eqs_1)
show(eqs_2)
show(eqs_3)

#colocando los coeficientes correctos y resolviendo

EQS_1 = desolve(eqs_1, psi, ivar=x)
A, B = var('A, B')
EQS_1 = EQS_1.subs(_K1=(A-B)*i,_K2=A+B)

EQS_2 = desolve(eqs_2, psi, ivar=x)

C, D = var('C, D')
EQS_2 = EQS_2.subs(_K1=C*i, _K2=C)

EQS_3 = desolve(eqs_3, psi, ivar=x)

E, F = var('E, F')

1
EQS_3 = EQS_3.subs(_K1=0, _K2=E)

#condiciones de continuidad

#E>V
EQS0_1 = EQS_1.subs(x=0)
EQS0_2 = EQS_2.subs(x=0)

EQSD_1=diff(EQS_1,x)
EQSD_2=diff(EQS_2,x)

EQSD_1 = EQSD_1.subs(x=0)
EQSD_2 = EQSD_2.subs(x=0)

#E<V
EQS0_3 = EQS_3.subs(x=0)
EQSD_3=diff(EQS_3,x)
EQSD_3 = EQSD_3.subs(x=0)

#Sistema de ecuaciones para hallar las constantes

eq1= EQS0_1==EQS0_2
eq2= EQSD_1==EQSD_2
sol = solve([eq1, eq2], B, C)
show(sol)

eq3= EQS0_1==EQS0_3
eq4= EQSD_1==EQSD_3
solc = solve([eq3, eq4], B, E)
#show(solc)

# Flujo y Coeficientes

#E>V

B= -A*(h-k)/(h+k)
C= 2*A*k/(h+k)

j_inc= hbar*k*(A*A.conjugate())/m_1
j_ref= hbar*k*(B*B.conjugate())/m_1
j_trs= hbar*h*(C*C.conjugate())/m_2

R = j_ref/j_inc
D = j_trs/j_inc

2
#E<V

B= -A*(j+i*k)/(j-i*k)
E= -2*i*A*k/(j-i*k)

j_incc= hbar*k*(A*A.conjugate())/m_1
j_refc= hbar*k*(B*B.conjugate())/m_1
j_trsc= hbar*j*(E*E.conjugate())/m_2

Rc = j_refc/j_incc
Dc = j_trsc/j_incc

k= sqrt(2*m_1*Ener)/hbar
h= sqrt(2*m_2*(Ener-V))/hbar
j= sqrt(2*m_2*(V-Ener))/hbar

R.subs(k= sqrt(2*m_1*Ener)/hbar,h= sqrt(2*m_2*(Ener-V))/hbar).show()

D.subs(k= sqrt(2*m_1*Ener)/hbar,h= sqrt(2*m_2*(Ener-V))/hbar).show()
Rc.subs(k= sqrt(2*m_1*Ener)/hbar,j= sqrt(2*m_2*(V-Ener))/hbar).show()
Dc.subs(k= sqrt(2*m_1*Ener)/hbar,j= sqrt(2*m_2*(V-Ener))/hbar).show()

#EQS0_1.show()
#EQS0_2.show()
#EQSD_1.show()
#EQSD_2.show()

∂2
k 2 ψ (x) + ψ (x) = 0
(∂x)2

∂2
h2 ψ (x) + ψ (x) = 0
(∂x)2

∂2
−j 2 ψ (x) + ψ (x) = 0
(∂x)2
[[ ]]
A(h − k) 2 Ak
B=− ,C =
h+k h+k

(√ √ √ √ )2
2 Em1 2 (E−V )m2
h̄ − h̄
(√ √ √ √ )2
2 Em1 2 (E−V )m2
h̄ + h̄

3
 √ √
8 Em1 (E − V )m2 m1
(√ √ √ √ )2
2 2 Em1 2 (E−V )m2
h̄ m2 h̄ + h̄

√ √
8 Em1 −(E − V )m2 m1
( √ √ √ √ )( √ √ √ √ )
2 Em1 2 −(E−V )m2 2 −(E−V )m2
2
h̄ m2 i
h̄ + h̄ − i 2 h̄ Em1 + h̄

[ ]:

[ ]:

[ ]:

[ ]:

[ ]:

[ ]:

[ ]:

4
5 Conclusiones
1. Se definieron e implementaron las fórmulas usadas para el cálculo de los problemas propuestos.

2. El programa Sagemath usado nos agilizó en la resolución y cálculo de los problemas propuestos.

6 Recomendaciones y/o perspectivas

1. Definir el signo de las variables en Sagemath, esto puede definir la convergencia de una integral.

7 Referencias
• D. Griffiths. Introduction to quamtum mechanics. 2005. Pearson, 47, 28.

• L. D. Landau and E. M. Lifshitz. Quantum Mechanics, volume 3. 1965. Addison-Wesley, 44.

23