使用scipy.integrate积分双球坐标函数遇TypeError错误求助

错误原因 #

• scipy.integrate.quad处理多变量积分时，会给被积函数传单个浮点数，你写的f函数里用enumerate(rp)和enumerate(thp)去迭代浮点数，直接触发TypeError: 'float' object is not iterable。
• 逻辑混淆：你需要对每个(r, th)点计算(rp, thp)的积分，但原代码把r_values、th_values直接嵌在被积函数里，导致变量关系混乱。
• 积分限错误：int(aperture_diameter / 2)把0.25转成0，int(2*np.pi)转成6，完全不符合实际积分范围。

修正后的完整代码 #

import numpy as np
from scipy.integrate import dblquad  # 用dblquad处理二重积分更直观

wavelength = 1e-3  # 单位：m
R = 5  # 半径，单位：m
aperture_diameter = 5e-1  # 单位：m
k = 2 * np.pi / wavelength  # 波数

# 定义(r, th)的采样点数量和取值范围
num_points = 100
r_values = np.linspace(0, aperture_diameter / 2, num_points)
th_values = np.linspace(0, 2 * np.pi, num_points)

# 被积函数：接收单个rp、thp浮点数，以及当前计算的r、th参数
def integrand(rp, thp, r, th):
    # 用三角恒等式简化角度差的余弦计算
    cos_angle_diff = np.cos(th - thp)
    # 计算相位项
    phase = -1j * k / (2 * R) * (r**2 + rp**2 - 2 * r**2 * rp**2 * cos_angle_diff)
    return rp * np.exp(phase)  # 极坐标积分必须乘rp（雅可比行列式）

# 存储每个(r, th)对应的积分结果，用复数类型存储
integral_results = np.zeros((num_points, num_points), dtype=np.complex128)

# 循环遍历每个(r, th)点，计算对应的二重积分
for i, r in enumerate(r_values):
    for j, th in enumerate(th_values):
        # dblquad参数说明：先内层积分变量（thp），再外层（rp）
        result, _ = dblquad(
            lambda thp, rp: integrand(rp, thp, r, th),
            0, aperture_diameter / 2,  # rp的积分范围
            lambda rp: 0, lambda rp: 2 * np.pi  # thp的积分范围（这里和rp无关，固定0到2π）
        )
        integral_results[i, j] = result

# 打印前3x3的结果示例
print("部分积分结果：")
print(integral_results[:3, :3])

核心修正说明 #

• 换用dblquad：quad主打单变量积分，dblquad专门处理二维积分，参数逻辑更贴合二重积分的需求。
• 被积函数适配单值输入：integrand只处理单个rp和thp值，不再尝试迭代数组，从根源避免类型错误。
• 分离循环与积分变量：把(r, th)作为参数传入被积函数，外层循环遍历所有(r, th)点，每个点单独计算积分，逻辑清晰。
• 修正积分限：去掉多余的int()转换，使用实际的浮点范围，保证积分区间正确。
• 简化角度计算：用cos(th - thp)替代原有的余弦正弦乘积和，代码更简洁高效。

内容的提问来源于stack exchange，提问作者Sophie Er