小编cam*_*ase的帖子

请原谅我以前问过这个问题，我找不到。我正在尝试使用向量运算将 numpy 数组逐步求和为新的 numpy 数组。我的意思是新数组的第二个索引等于旧数组的第一个+第二个索引。或 A[n] = B[0] + B[1] ... + B[n]。我知道如何使用 for 循环来执行此操作，但我正在寻找矢量化解决方案。

这是我的非矢量化解决方案：

import numpy as np  
A = np.arange(10)
B = np.empty(10)

for i in range(len(A)):
    B[i] = sum(A[0:i+1])

print(B)