python中shift函数_pandas DataFrame.shift()函数

pandas DataFrame.shift()函数可以把数据移动指定的位数

period参数指定移动的步幅,可以为正为负.axis指定移动的轴,1为行,0为列.

eg: 有这样一个DataFrame数据:

import pandas as pd

data1 = pd.DataFrame({

'a': [0, 1, 2, 3, 4, 5, 6, 7, 8, 9],

'b': [9, 8, 7, 6, 5, 4, 3, 2, 1, 0]

})

print data1

a b

0 0 9

1 1 8

2 2 7

3 3 6

4 4 5

5 5 4

6 6 3

7 7 2

8 8 1

9 9 0

如果想让 a和b的数据都往下移动一位:

data2 = data1.shift(axis=0)

print data2

a b

0 NaN NaN

1 0.0 9.0

2 1.0 8.0

3 2.0 7.0

4 3.0 6.0

5 4.0 5.0

6 5.0 4.0

7 6.0 3.0

8 7.0 2.0

9 8.0 1.0

如果是在行上往右移动一位:

data3 = data1.shift(axis=1)

print data3

a b

0 NaN 0.0

1 NaN 1.0

2 NaN 2.0

3