Python实现随波长变化的光学传输矩阵乘法报错解决方案

触发报错的核心原因 #

• 构造L2矩阵时误用了cmath.exp接口：该函数是Python标准库的标量计算接口，仅能接收单个复数标量作为输入。当前代码中psi是由波长序列lam计算得到的一维numpy数组，把数组传入标量接口就会触发“仅长度1数组可转换为Python标量”的类型错误。
• 原代码的L2构造逻辑不符合numpy批量矩阵运算的格式要求：需要生成的是覆盖所有波长采样点的2×2矩阵数组（整体为三维结构，第一维对应不同波长点，后两维为对应点的2×2矩阵数值），嵌套Python列表的结构无法直接执行批量矩阵乘法。

逐点修正代码逻辑 #

1. 替换计算接口：把所有针对数组的复数指数运算从cmath.exp换成numpy原生的np.exp，该接口支持对数组逐元素做复数运算，完全适配序列计算场景。
2. 按numpy批量乘法要求的格式构造L2介质传输矩阵：初始化形状匹配的三维复数数组，填充对角项数值：

   # L2数组维度：[波长采样点索引, 矩阵行, 矩阵列]，适配numpy广播规则
   L2 = np.zeros((len(lam), 2, 2), dtype=np.complex128)
   # 填充介质层传输矩阵的对角相位项
   L2[:, 0, 0] = np.exp(-1j * psi)
   L2[:, 1, 1] = np.exp(1j * psi)
   

3. 替换固定界面矩阵为numpy数组并适配批量乘法：把原代码中嵌套Python列表形式的I12、I23转换成numpy数组，利用numpy的广播特性直接计算全波长下的总传输矩阵：

   # 转换固定界面矩阵为numpy数组
   I12_np = np.array([[1, r12], [r12, 1]])
   I23_np = np.array([[1, r23], [r23, 1]])
   coeff = (1 / t12) * (1 / t23)
   # 逐波长点计算三个矩阵的乘积，结果为(N,2,2)的三维数组，每个波长点对应一个总传输矩阵
   M_total = coeff * (I12_np @ L2 @ I23_np)
   

4. 提取指定矩阵分量完成绘图：根据作业要求提取总矩阵的目标分量（比如透射率、反射率对应的矩阵元），以lam为X轴、分量的计算值为Y轴即可生成要求的波长-矩阵分量关系曲线。

实操注意：凡是涉及numpy数组的逐元素数学运算，优先使用numpy内置的向量化接口，不要混用Python标准库的math、cmath模块，这类模块仅支持标量输入，遇到数组参数必然触发类型错误。

内容的提问来源于stack exchange，提问作者Min-Gyo Jeong