Pandas使用shift函数执行移位操作时如何保留原数据最后一行？

实现方法 #

pd.DataFrame.shift() 默认返回与原数据长度一致的结果，移位后原始数据的最后一行会被移出结果范围，你可以通过以下两种方式实现需求：

首先构造测试数据：

import pandas as pd
df = pd.DataFrame({'A': [1.0, 2.0, 3.0]})

方法1：拼接移位结果与原数据末尾行（多列通用）

直接将移位后的结果和原数据的最后1行拼接，重置索引即可得到目标结果：

res = pd.concat([df.shift(1), df.tail(1)], ignore_index=True)

方法2：先扩展索引再填充

先将结果的索引长度扩展为原数据长度+1，再填充移位后的值，最后一位补原数据的最后一个值：

# 先扩展索引
res = df.reindex(range(len(df) + 1))
# 填充移位值，最后一位补原数据末尾值
res['A'] = df['A'].shift(1).reindex(res.index, fill_value=df['A'].iloc[-1])

两种方法输出的结果均为：

A
0  NaN
1  1.0
2  2.0
3  3.0

内容的提问来源于stack exchange，提问作者Fatafim