在 Numpy Arrays Python

问题描述

如何在不使用 for 循环的情况下编写一个添加到数组的前一个索引的函数。因此，对于预期输出中的第二个值，它将是前一个索引 (2) 加上当前索引 (5) 导致 7。我怎么能做这样的事情？

import numpy as np

A = np.array([2,5,44,-12,3,-5])

预期输出：

[2,7,51,39,42,37]

解决方法

给你：

np.cumsum(A)

打印：

array([ 2,7,51,39,42,37],dtype=int32)