Получить скользящее среднее в фрейме данных pyspark

У меня есть блок данных искры с несколькими столбцами. Мне нужно получить скользящее среднее и перемещение стандартного отклонения для конкретного столбца и добавить его к кадру данных.

В качестве примера:

col1   col2

1       5

1       10

1       15

и если мое окно для перемещения означает 2, тогда мой df должен выглядеть так:

col1     col2     mean
1        5        5
1        10       7.5
1        15       12.5

Я нашел один способ, то есть преобразовать его в pandas dataframe, использовать для этого свою функцию.

Но я хочу сделать это в pyspark dataframe, так как мои данные могут быть очень большими.