Pythonでcsvファイルの前後の行から平均を取りたい

Question

前提・実現したいこと

初めまして、Python初学者です。
CSVファイルをを読み込んで、とある列のあるセルを基準として前後５個（基準とその前４個、後５個）計１０個の数値データの平均を取り、別の列に代入したいと考えているのですが、基準から前後５個をどう取得すればいいのか分からず困っています。

データの形は例えば

Data1
2
2
1
3
2
1
2
1
2
1
2

となっていた時、

Data1	Ave
2	NaN
2	NaN
1	NaN
3	NaN
2	1.7
1	1.7
2	NaN
1	NaN
2	NaN
1	NaN
2	NaN

となるように新たにAveという列を作成し、値を代入したいと思っています。

このように計算したい場合、どのように記述すればいいのかご教授いただけますと幸いです。
よろしくお願いいたします。

Comments

[melian]

Ave が 1.9 になっている行は 1.7 ではないでしょうか？

[rinrinapple]

失礼致しました、おっしゃる通り正しくは1.7でした

Answer 1

import pandas as pd

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

df['Ave'] = df.rolling(10, center=True).mean().shift(-1)

print(df)

_ = '''
    Data1  Ave
0       2  NaN
1       2  NaN
2       1  NaN
3       3  NaN
4       2  1.7
5       1  1.7
6       2  NaN
7       1  NaN
8       2  NaN
9       1  NaN
10      2  NaN
'''

Answer 2

pandasのDataFrameを使うと以下です。

>>> print(df)
    Data1
0       2
1       2
2       1
3       3
4       2
5       1
6       2
7       1
8       2
9       1
10      2
>>>
>>> df['Ave'] = df['Data1'].rolling(10).mean().shift(-5)
>>> print(df)
    Data1  Ave
0       2  NaN
1       2  NaN
2       1  NaN
3       3  NaN
4       2  1.7
5       1  1.7
6       2  NaN
7       1  NaN
8       2  NaN
9       1  NaN
10      2  NaN

詳しくはpandasで窓関数を適用するrollingを使って移動平均などを算出をお読みください。
また、csvファイルを読み込んで書き出したいなら、pandasでcsv/tsvファイル読み込み（read_csv, read_table）とpandasでcsvファイルの書き出し・追記（to_csv）を使ってください。