pandasのfor文中での前後行参照

Question

pandasのDataFrameにて，for文で行ごとに処理するとき，
前後の数行を参照したいときがあります．
何か方法はありますか．

Comments

[tachikoma]

どういう処理をやりたいのかソースコードで示して下さい。擬似コードでもいいです。

Answer 1

ループ処理を行う前に予め DataFrame.shift() 使って、前後の行のデータをコピーした列データを作成しておけば良いのでは。

import pandas as pd

df = pd.DataFrame({'val': [1,2,3,4,5]})

df['prev_val'] = df['val'].shift(1)
df['next_val'] = df['val'].shift(-1)

print(df)
#   val  prev_val  next_val
#0    1       NaN       2.0
#1    2       1.0       3.0
#2    3       2.0       4.0
#3    4       3.0       5.0
#4    5       4.0       NaN