如何使用 scipy.integrate 库集成此功能？

Question

所以我编写了这个公式我得到：def sumAN(theta,CoefAN,n_cl):    # this function give us the sumatory in the right side of the formula    Sumatorio = np.array([])    for count,i in enumerate(theta):        sumatorio = 0        for count2,j in enumerate(n_cl):            sumatorio = sumatorio +CoefAN[count2]*sin(int(count2+1)*i)        Sumatorio = np.append(Sumatorio,sumatorio)    return Sumatoriocl= 4*((np.radians(alpha)+A0)*tan(theta/2)+sumAN(theta,CoefAN,n_cl))稍微解释一下：- Alpha：常量- A0：常量- AN ：np.array([])(n 值)- theta：自变量在此之后，我需要计算下一个积分：这是我遇到问题的地方：ch = integrate.quad(lambda theta:(4*((alpha_char+A0)*tan(theta/2)+sumAN(theta,CoefAN,n_charl)))*(cos(theta)-cos(xa))*sin(theta),0,xa)[0]我有所有的限制和一切。但我得到下一个错误：“浮动”对象不可迭代我不知道如何继续。所以我的问题是：如何使用integrate.quad 方法集成这个函数？也许我会改变总结的方式？我如何以其他方式编写函数？提前致谢

Answer 1

这应该工作

import numpy as np

from scipy.integrate import quad

def integrand(theta, theta_a, alpha, A):

    sum = 0

    # get sum

    for index, value in enumerate(A):

        if index == 0:

            sum += (alpha + A[index]) * np.tan(0.5 * theta)

        else:

            sum += A[index] * np.sin(index * theta)

    # note that multiplication with 4 and multiplication with 1/4

    # result in one as prefactor

    return -sum * (np.cos(theta) - np.cos(theta_a))

# calculate integral 

theta_a = 0

alpha = 0 

array_coefficients = np.array([1, 2, 3, 4])

integral = quad(integrand, 0, 1, args=(theta_a , alpha, array_coefficients))