给定一个大小为N的列表a，我如何创建一个列表B，使得B的每个元素都是a和B的前一个元素的和？

Question

我试图创建一个函数my_function(a)，它接受一个由N个数字组成的列表a，并输出一个由N+1个数字组成的列表B，B[0]=0和B[N]=B[n-1]+a[n-1]，其中N

例如，如果A是列表[1,2,3,4,5]，B[1]=B[0]+A[0]=0+1,B[2]=B[1]+A[1]=1+2，则最终输出为[0,1,3,6,10,15]

我最多只需要使用numpy。

我的密码是

def my_function(A):
   for n in range(len(A)):
      B[0]=0
      B[n]=B[n-1]+A[n-1]
      return B[n]

我找不到没有错误地定义B的方法

Answer 1

您只需要正确地初始化B：

def my_function(A):
    B = [0] * (len(A) + 1) # list with N + 1 Zeros
    for n in range(1, len(A) + 1): # n in 1..N
        B[n] = B[n-1] + A[n-1]
    return B

或

def my_function(A):
    B = [0] * (len(A) + 1) # list with N + 1 Zeros
    for n in range(len(A)): # n in 0..N-1
        B[n+1] = B[n] + A[n]
    return B

python使用itertools.accumple:

from itertools import accumulate

def my_function(A):
    return [0, *accumulate(A)] # accumulate doesn't start with 0