Используя numpy, как я могу сгенерировать массив, в котором значение в каждом индексе является суммой значений от 0 до того же индекса во втором массиве?

Question

С учетом массива numpy a = [3,5,7]

Как эффективно сгенерировать второй массив b, где b [i] = numpy.Sum (a [0: i]?

Я просмотрел множество документов, но решение не бросается в глаза ...

Ожидаемый результат будет b = [3,8,15]

Любые идеи будут с благодарностью приняты !!!

Спасибо,

Дуг

Answer 1

Похоже, вам нужна функция cumsum от numpy:

a=np.array([3,5,7])

In [1]: np.cumsum(a)
Out[1]: array([ 3,  8, 15], dtype=int32)