请补充上面的代码让它输出结果绘图：粒子以不同能量经过方势阱时，波函数的变化情况

可以使用以下代码来实现粒子在方势阱中的传播和波函数的绘制：

import matplotlib.pyplot as plt

# 计算波函数
def calc_wave_func(k, x, N):
    mat_V = square_poten_well(x, N)
    G = Green_func(k, x, x, N)
    G_inv = np.linalg.inv(G - mat_V)
    phi_k = phi(k, x, N)
    return np.dot(G_inv, phi_k)

# 绘制波函数
def plot_wave_func(k, x, N):
    wave_func = calc_wave_func(k, x, N)
    plt.plot(x, np.abs(wave_func))
    plt.title(f'Wave function for k={k}')
    plt.xlabel('x')
    plt.ylabel('|psi(x)|')
    plt.show()

# 绘制不同能量下的波函数
for k in k_vec:
    plot_wave_func(k, x, N)

这段代码中，`calc_wave_func()` 函数用于计算特定波数下的波函数，其中使用了 `np.linalg.inv()` 函数来计算格林函数的逆矩阵，从而得到波函数。`plot_wave_func()` 函数用于绘制特定能量下的波函数，其中使用了 `np.abs()` 函数将波函数转换为幅度。最后，使用循环对不同能量下的波函数进行绘制。

运行以上代码，即可得到粒子在不同能量下通过方势阱时，波函数的变化情况的绘图。