sympy 求微分方程_零七：常微分方程，埃里函数\\附一些关于plt作图

#sympy常微分方程：sympy.dsolve(Eq,f,n,ics)
#Eq:方程，f:函数，n:阶数（幂级数解），ics:初始条件（边界）
#埃里函数：sympy.airyai和sympy.airybi

import sympy
import matplotlib.pyplot as plt

f=sympy.Function('f')
x=sympy.Symbol('x',real=True)
print(sympy.latex(sympy.dsolve(f(x).diff(x,2)-x*f(x),f(x)))) #线性势

print(sympy.latex(sympy.dsolve(f(x).diff(x,2)-2*x*f(x).diff(x)+f(x),f(x),n=9)))
#不会截断的谐振子
print(sympy.latex(sympy.dsolve(f(x).diff(x,2)-2*x*f(x).diff(x)+2*f(x),f(x),n=9)))
#奇宇称截断
print(sympy.latex(sympy.dsolve(f(x).diff(x,2)-2*x*f(x).diff(x)+4*f(x),f(x),n=9)))
#偶宇称截断

x=[_*0.02 for _ in range(1001)]
y=list(map(lambda k:float(sympy.airyai(-k)),x))
z=list(map(lambda k:float(sympy.airyai(k-9.02265088)**2),x))

ax = plt.gca()
ax.spines['right'].set_color('none') #去掉右轴
ax.spines['top'].set_color('none')
ax.xaxis.set_ticks_position('bottom') #将下轴设为x轴
ax.yaxis.set_ticks_position('left')          
ax.spines['bottom'].set_position(('data', 0)) #绑定0点
ax.spines['left'].set_position(('data', 0))
plt.plot(x,y)
plt.show()

plt.cla()
plt.plot(x,z)
plt.show()

Ai(-x)，用于求零点，能级……绑定了x轴y轴的零点

与WKB近似算得的零点位置对照：

第六个可取能级（第五激发态）的波函数平方（未归一化）

第35个能级