numpy数组元素的累积整合

时间:2015-09-01 20:53:05

标签: python arrays numpy scipy integration

我想进行以下类型的整合:

说我有2个阵列

a = np.array[1,2,3,4]
b = np.array[2,4,6,8]

我知道如何使用以下内容集成这些内容:

c = scipy.integrate.simps(b, a)

上述数据集的c = 15

我想要做的是将每个数组的第一个元素相乘并添加到名为d的新数组,即a[0]*b[0]然后将前2个元素与数组集成,然后将前3个元素集成在一起,等等所以最终对于这个数据集,我会得到

d = [2 3 8 15]

我尝试了一些但没有运气;我是编写代码的新手。

1 个答案:

答案 0 :(得分:1)

如果我已正确理解您的需求,您可以执行以下操作:

import numpy as np
from scipy import integrate

a = np.array([2,4,6,8])
b = np.array([1,2,3,4])
d = np.empty_like(b)
d[0] = a[0] * b[0]
for i in range(2, len(a) + 1):
    d[i-1] = integrate.simps(b[0:i], a[0:i])
print(d)