数值积分循环Python

Question

我想编写一个程序，在循环中解决下面的定积分，该循环考虑每次迭代的常数c的不同值。

然后我想将积分的每个解决方案输出到一个新数组中。

如何在python中最好地编写这个程序？

，限制在0到1之间。

from scipy import integrate

integrate.quad

这里可以接受。我的主要斗争是构建该计划。

这是一次旧尝试（失败）

# import c
fn = 'cooltemp.dat'
c = loadtxt(fn,unpack=True,usecols=[1])

I=[]
for n in range(len(c)):

    # equation
    eqn = 2*x*c[n]

    # integrate 
    result,error = integrate.quad(lambda x: eqn,0,1)

    I.append(result)

I = array(I)

Answer 1

例如，计算[0,9]中c的给定积分：

[scipy.integrate.quadrature(lambda x: 2 * c * x, 0, 1)[0] for c in xrange(10)]

这是使用list comprehension和lambda functions。

或者，您可以定义函数，该函数将给定c中的积分作为ufunc返回（感谢vectorize）。这可能更多是在numpy的精神。

>>> func = lambda c: scipy.integrate.quadrature(lambda x: 2 * c * x, 0, 1)[0]
>>> ndfunc = np.vectorize(func)
>>> ndfunc(np.arange(10))
array([ 0.,  1.,  2.,  3.,  4.,  5.,  6.,  7.,  8.,  9.])

Answer 2

你真的很亲密。

fn = 'cooltemp.dat'
c_values = loadtxt(fn,unpack=True,usecols=[1])

I=[]
for c in c_values: #can iterate over numpy arrays directly.  No need for `range(len(...))`

    # equation
    #eqn = 2*x*c[n] #This doesn't work, x not defined yet.

    # integrate 
    result,error = integrate.quad(lambda x: 2*c*x, 0, 1)

    I.append(result)

I = array(I)

我认为你对lambda的工作方式感到有些困惑。

my_func = lambda x: 2*x

与以下内容相同：

def my_func(x):
    return 2*x

如果您仍然不喜欢lambda，可以这样做：

f(x,c):
   return 2*x*c

#...snip...
integral, error = integrate.quad(f, 0, 1, args=(c,) )

Answer 3

constants = [1,2,3]

integrals = []                                  #alternatively {}

from scipy import integrate

def f(x,c):
    2*x*c

for c in constants:
    integral, error = integrate.quad(lambda x: f(x,c),0.,1.)
    integrals.append(integral)                 #alternatively integrals[integral]

这将输出一个类似Nicolas答案的列表，用于任何常量列表。