最后一行代码，出错，找不到替代的方法


import numpy as np
import pandas as pd
#from pandas import DataFrame

#赋值“A”列：
df = pd.DataFrame(np.ones(100),columns=['A'])


#赋值“B”列：
df['B'] = pd.DataFrame(np.random.randint(0,15,100))


#打印两列
print(df[['A','B']])


#定义函数
def SUM(S, N):            #对序列求N天累计和，返回序列    N=0返回零        
    return pd.Series(S).rolling(N).sum().values if N>0 else 0




#调试出错
df.loc[:,'C'] = SUM(df['A'],df['B'])

#SUM(S, N) 函数中，参数N，要求固定值，但我需要 N 依次取 'B'列所有100个数值，求实现，谢谢。
--
FROM 211.161.241.*