Python Pandas DataFrame shift方法用法及代码示例

例子

考虑以下 DataFrame ：

df = pd.DataFrame({"A":["a","b","c","d"],"B":["e","f","g","h"],"C":["i","j","k","l"]})
df



   A  B  C
0  a  e  i
1  b  f  j
2  c  g  k
3  d  h  l

基本用法

向下移动 2 行：

df.shift(2)



   A    B    C
0  NaN  NaN  NaN
1  NaN  NaN  NaN
2  a    e    i
3  b    f    j

向上移动 2 行：

df.shift(-2)



   A    B    C
0  c    g    k
1  d    h    l
2  NaN  NaN  NaN
3  NaN  NaN  NaN

请注意索引没有移动，即索引仍然是 [0,1,2,3] 。

移动列

默认情况下， axis=0 ，这意味着行发生移位。要移动列，请像这样传递axis=1：

df.shift(2, axis=1)



   A    B    C
0  NaN  NaN  a
1  NaN  NaN  b
2  NaN  NaN  c
3  NaN  NaN  d

再次注意列值的变化，但不注意列标签的变化。

指定fill_value

我们可以指定我们自己的填充值，而不是使用默认的 NaN 填充，如下所示：

df.shift(fill_value=0)



   A  B  C
0  0  0  0
1  0  0  0
2  a  e  i
3  b  f  j

指定频率

当 DataFrame 是时间序列时，freq 参数发挥作用。

为了进行演示，请考虑以下时间序列 DataFrame：

index_date = pd.date_range("2020-12-25", periods=4, freq="D")
df = pd.DataFrame({"A":["a","b","c","d"],"B":["e","f","g","h"],"C":["i","j","k","l"]}, index=index_date)
df



            A  B  C
2020-12-25  a  e  i
2020-12-26  b  f  j
2020-12-27  c  g  k
2020-12-28  d  h  l

要将索引移动 3 天：

df.shift(freq="3D")



            A  B  C
2020-12-28  a  e  i
2020-12-29  b  f  j
2020-12-30  c  g  k
2020-12-31  d  h  l

请注意索引中的日期已移动 3 天，但值(例如 a 、 b 等)根本没有移动。

您还可以组合 period 和 freq 。要将索引移动 2*3=6 days ：

df.shift(2, freq="3D")



            A  B  C
2020-12-31  a  e  i
2021-01-01  b  f  j
2021-01-02  c  g  k
2021-01-03  d  h  l

不直观的行为

考虑以下 DataFrame ：

index_date = pd.date_range("2020-12-29", periods=4, freq="D")
df = pd.DataFrame({"A":["a","b","c","d"],"B":["e","f","g","h"],"C":["i","j","k","l"]}, index=index_date)
df



            A  B  C
2020-12-29  a  e  i
2020-12-30  b  f  j
2020-12-31  c  g  k
2021-01-01  d  h  l

设置 freq="M" 会产生：

df.shift(periods=1, freq="M")



            A  B  C
2020-12-31  a  e  i
2020-12-31  b  f  j
2021-01-31  c  g  k
2021-01-31  d  h  l

结果有点不直观，因为我们期望将 one-month 偏移量应用于每个日期索引。

请注意以下事项：

• 每个日期索引都转移到月底。

• 由于我们指定了 freq="M" ，因此日期索引的日单位变得无关紧要。

• 例外情况是当日单位位于月末时(例如 12-31 )，在这种情况下，月份将发生偏移(例如 01-31 )。