python绘制衍射全息图的效果图

Python中可以使用一些科学计算库如NumPy、Matplotlib以及专门用于图像处理的Pillow等工具来绘制衍射全息图。全息图通常涉及干涉图案，需要创建两个波前的迭加效果来模拟实际物理过程。

以下是一个简单的示例，展示如何使用这些库来创建二维衍射图案：

import numpy as np
import matplotlib.pyplot as plt
from PIL import Image

# 创建光源波函数和物体波函数
def create_wave_function(size, wavelength=1):
    x, y = np.meshgrid(np.arange(size), np.arange(size))
    return np.sin(2 * np.pi / wavelength * x)

# 创建全息图，通过迭加两波函数
def hologram(primary, secondary):
    return primary + secondary

# 设置参数
size = 500
wavelength = 600  # 假设红光波长
primary = create_wave_function(size)
secondary = create_wave_function(size, wavelength=400)  # 另一波长假设蓝光

# 计算全息图
hologram_img = hologram(primary, secondary)

# 将数组转换为图像并显示
plt.imshow(hologram_img, cmap='gray')
plt.title('衍射全息图')
plt.colorbar()
plt.show()