Усреднение значения на основе предыдущего значения Python

Question

У меня есть датафрейм, похожий на:

value 1         value 2
1               10
4               1
5               8
6               10
10              12

Я хочу перейти к каждой записи value 1, усреднить предыдущее значение, а затем создать новый столбец рядом с value 2 со средним значением.

Вывод должен выглядеть следующим образом:

value 1         value 2        avg
1               10             nan
4               1              2.5
5               8              4.5 
6               10             5.5
10              12             8.0
.
.

Как бы я поступил так?

Answer 1

shift

Вы можете суммировать серию со сдвинутой версией самого себя:

df['avg'] = (df['value1'] + df['value1'].shift()) / 2

print(df)

   value1  value2  avg
0       1      10  NaN
1       4       1  2.5
2       5       8  4.5
3       6      10  5.5
4      10      12  8.0