在没有 for 循环的情况下迭代 scipy.integrate 中的 args 数组

如何解决在没有 for 循环的情况下迭代 scipy.integrate 中的 args 数组

假设我有一个函数 f(x) = a*x，我想对给定的 a 对 x 进行积分。我希望能够通过 scipy.integrate.quad 一个 arg 'a' 数组来迭代。对于一个参数，我知道这可以通过

完成

import numpy as np
from scipy import integrate
param = 1 
def f(x,a):
    return a*x
integral = integrate.quad(f,1,args = (param)) # 0,1 are just arbitrary limits of integration
print(integral)

按预期工作。但是，如果我想遍历参数值数组，例如。）

param = np.array([1,2])
def f(x,args = (param))
print(integral)

我收到错误消息：“只能将大小为 1 的数组转换为 Python 标量”。显然我可以通过实现类似的东西来避免这种情况

integrals = []
for i in param:
    integrals.append(integrate.quad(f,args = (i)))

但我想尽可能避免 for 循环，以便为更大尺寸的参数数组和多重积分保持代码快速。你建议我做什么？

解决方法

试试 quad_vec：https://docs.scipy.org/doc/scipy/reference/generated/scipy.integrate.quad_vec.html

同样不要忘记参数元组的逗号：args=(i,)