《量子力学教程》曾谨言著，第二章课后习题作业2.4画图题

作图，比较 $\psi (x)$ 与 ${\psi }_1(x)$ 曲线.从 $P_1\gg P_n(n\ne 1)$ 来说明两条曲线非常相似，即 $\psi (x)$ 几乎与基态 ${\psi }_1(x)$ 完全相同.

# 量子力学作图
import numpy as np

import matplotlib.pyplot as plt


plt.rcParams['font.sans-serif'] = ['SimHei']  # 指定默认字体
plt.rcParams['axes.unicode_minus'] = False  # 解决保存图像是负号'-'显示为方块的问题

x = np.linspace(0,1,20)
a = 1
psi = np.sqrt(30/a**5)*x*(x-a)
psi_1 = -np.sqrt(2/a)*np.sin(np.pi*x/a)

plt.plot(x,psi,'--',label=r'$\psi(x)$')
plt.plot(x,psi_1,'ro',label=r'$\psi_1(x)$')
plt.legend()
plt.title('两条曲线的比较')

结果：