Python瑞利-索末菲衍射公式

 1. 代码

import numpy as np
import matplotlib.pyplot as plt

# 定义常量
wavelength = 500e-9  # 波长 (500 nm)
k = 2 * np.pi / wavelength  # 波数

# 定义源点和观察点的坐标
z = 0.1  # 观察屏与孔的距离 (m)
x = np.linspace(-0.01, 0.01, 500)  # 观察屏的横坐标范围 (m)
y = np.linspace(-0.01, 0.01, 500)  # 观察屏的纵坐标范围 (m)
X, Y = np.meshgrid(x, y)

# 定义孔的尺寸
a = 1e-3  # 孔的半径 (m)

# 计算衍射场
U = np.zeros_like(X, dtype=np.complex128)  # 初始化衍射场

for i in range(X.shape[0]):
    for j in range(X.shape[1]):
        r = np.sqrt(X[i, j]**2 + Y[i, j]**2 + z**2)  # 观察点到源点的距离
        if np.sqrt(X[i, j]**2 + Y[i, j]**2) <= a:
            U[i, j] = (np.exp(1j * k * r) / r) * (1 + 1j / (k * r))

# 计算强度
I = np.abs(U)**2

# 可视化结果
plt.figure(figsize=(8, 6))
plt.imshow(I, extent=[x.min(), x.max(), y.min(), y.max()], cmap='inferno')
plt.colorbar(label='Intensity')
plt.xlabel('x (m)')
plt.ylabel('y (m)')
plt.title('Rayleigh-Sommerfeld Diffraction Pattern')
plt.show()

2. 结果