```python import numpy as np # 定义势能函数 def potential_energy(x, V0): return V0 * np.exp(-x**2 / (2 * V0)) # 定义初始条件 initial_position = 1.0 initial_momentum = 0.0 initial_energy = 1.0 # 计算波函数 def wave_function(x, t, E0, V0): psi = np.zeros((V0, 2)) for i in range(V0): psi[i] = (np.cos(E0 * t) + E0 * initial_momentum * np.sin(E0 * t)) / (E0 ** 3) return psi # 初始化参数 V0 = 1.0 E0 = 1.0 V0_potential = 1.0 V0_initial_momentum = 1.0 V0_initial_energy = 1.0 # 计算时间步长 dt = 0.01 t = np.linspace(0, 1, int(dt * V0), endpoint=False) # 计算能量和波函数 E = 0.5 * V0_initial_energy psi = wave_function(initial_position, t, E, V0_potential) print("波函数:") print(psi) ```