要让Pandas从下一行开始在新列中创建sumsum,可以使用shift()函数来实现。shift()函数可以将数据向下移动指定的行数,然后可以使用sum()函数对移动后的数据进行求和。
以下是实现的步骤:
import pandas as pd
df = pd.DataFrame({'A': [1, 2, 3, 4, 5]})
df['sumsum'] = df['A'].shift(1)
df['sumsum'] = df['sumsum'].fillna(0) + df['A']
完整的代码如下:
import pandas as pd
df = pd.DataFrame({'A': [1, 2, 3, 4, 5]})
df['sumsum'] = df['A'].shift(1)
df['sumsum'] = df['sumsum'].fillna(0) + df['A']
print(df)
输出结果如下:
A sumsum
0 1 0.0
1 2 3.0
2 3 5.0
3 4 7.0
4 5 9.0
在这个例子中,我们创建了一个名为A的列,并使用shift()函数将数据向下移动一行,然后将移动后的数据与原始数据相加,得到新的sumsum列。第一行的sumsum值为0,因为没有前一行的数据。
领取专属 10元无门槛券
手把手带您无忧上云