请修改这段代码让它输出结果绘图:粒子以不同能量经过方势阱时,波函数的变化情况:import numpy as np def square_poten_well(x, N): L = 2 V0 = -1 mat_V = np.zeros((N, N)) for i, xx in enumerate(x): if abs(xx) <= L/2: mat_V[i, i] = V0 return mat_V def phi(k, x, N): return [np.exp(1.0j*k*x[i]) for i in range(N)] def Green_func(k, x, xp, N): G = np.ones((N, N), dtype=complex) for i in range(N): G[i, :] = [-1.0j / k * np.exp(1.0j*k*np.abs(x[i]-xp[j])) for j in range(N)] return G def change_of_var(node, weight, a, b, N): nop = [(b-a) * node[i] / 2.0 + (a+b) / 2.0 for i in range(N)] wp = [(b-a) / 2.0 * weight[i] for i in range(N)] return nop, wp N = 298 a = -1.5 b = 1.5 k_vec = np.arange(0.5, 6.0) x, w = np.polynomial.legendre.leggauss(N) x = (b-a)/2.0*x + (b+a)/2.0 w = (b-a)/2.0*w
时间: 2024-03-08 22:46:06 浏览: 32
以下是修改后的代码,其中添加了绘图部分:
```python
import numpy as np
import matplotlib.pyplot as plt
def square_poten_well(x, N):
L = 2
V0 = -1
mat_V = np.zeros((N, N))
for i, xx in enumerate(x):
if abs(xx) <= L/2:
mat_V[i, i] = V0
return mat_V
def phi(k, x, N):
return [np.exp(1.0j*k*x[i]) for i in range(N)]
def Green_func(k, x, xp, N):
G = np.ones((N, N), dtype=complex)
for i in range(N):
G[i, :] = [-1.0j / k * np.exp(1.0j*k*np.abs(x[i]-xp[j])) for j in range(N)]
return G
def change_of_var(node, weight, a, b, N):
nop = [(b-a) * node[i] / 2.0 + (a+b) / 2.0 for i in range(N)]
wp = [(b-a) / 2.0 * weight[i] for i in range(N)]
return nop, wp
N = 298
a = -1.5
b = 1.5
k_vec = np.arange(0.5, 6.0)
x, w = np.polynomial.legendre.leggauss(N)
x = (b-a)/2.0*x + (b+a)/2.0
w = (b-a)/2.0*w
# 绘制波函数变化图像
fig, axes = plt.subplots(nrows=2, ncols=3, figsize=(12, 8))
for i, k in enumerate(k_vec):
wave_func = np.abs(np.array(calc_wave_func(k, x, N)))
ax = axes[i//3, i%3]
ax.plot(x, wave_func)
ax.set_title(f'Wave function for k={k:.1f}')
ax.set_xlabel('x')
ax.set_ylabel('|psi(x)|')
plt.tight_layout()
plt.show()
```
这段代码中,我们添加了绘图部分,其中使用了 `subplots()` 函数创建了 2x3 的图像区域,然后使用循环对不同能量下的波函数进行绘制,最后使用 `tight_layout()` 函数调整图像布局,使得图像之间的间隔更加美观。
相关推荐
![zip](https://img-home.csdnimg.cn/images/20210720083736.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)