将数据从第1行移动到第0行

Question

我用python编写了这个函数。我想让这个东西显示行与生产列之间的差异

下面是代码

def print_df():
mycursor.execute("SELECT * FROM productions")

myresult = mycurson.fetchall()
myresult.sort(key=lambda x: x[0])

df = pd.DataFrame(myresult, columns=['Year', 'Production (Ton)'])
df['Dif'] = abs(df['Production (Ton)']. diff())
print(abs(df))

当然输出是这个

Year Production (Ton)       Dif
0   2010           339491       NaN
1   2011           366999   27508.0
2   2012           361986    5013.0
3   2013           329461   32525.0
4   2014           355464   26003.0
5   2015           344998   10466.0
6   2016           274317   70681.0
7   2017           200916   73401.0
8   2018           217246   16330.0
9   2019           119830   97416.0
10  2020            66640   53190.0

但是我想要这样的输出

Year Production (Ton)       Dif
0   2010           339491   27508.0    
1   2011           366999    5013.0
2   2012           361986   32525.0 
3   2013           329461   26003.0
4   2014           355464   10466.0
5   2015           344998   70681.0
6   2016           274317   73401.0
7   2017           200916   16330.0
8   2018           217246   97416.0
9   2019           119830   53190.0
10  2020            66640   66640.0

我应该改变或添加什么到我的代码?

Answer 1

您可以使用diff的负周期输入来获得您想要的差异，然后fillna用Production列的值填充最后一个值:

df['Dif'] = df['Production (Ton)'].diff(-1).fillna(df['Production (Ton)']).abs()

输出:

Year  Production (Ton)      Dif
0   2010            339491  27508.0
1   2011            366999   5013.0
2   2012            361986  32525.0
3   2013            329461  26003.0
4   2014            355464  10466.0
5   2015            344998  70681.0
6   2016            274317  73401.0
7   2017            200916  16330.0
8   2018            217246  97416.0
9   2019            119830  53190.0
10  2020             66640  66640.0

Answer 2

使用shift(-1)将所有行上移一个位置

df['Dif'] = (df['Production (Ton)'] - df['Production (Ton)'].shift(-1).fillna(0)).abs()

注意，通过设置fillna(0)，您可以避免nan。

也可以使用diff:

df['Dif'] = df['Production (Ton)'].diff().shift(-1).fillna(0).abs()