(Python): скользящее среднее с использованием 2D-массива

Question

У меня есть 2D-список, и я хочу вычислить скользящее среднее по номерам столбцов. У меня есть следующий код. Кто-нибудь знает метод numpy, который возвращает новый 2D-список со скользящей средней всех столбцов?

v = [[1, 2, 3],
     [4, 5, 6],
     [7, 8, 9]]

print((v[0][0]+v[1][0])/2)

Возвращает

2.5

Answer 1

IIU C и у вас нет проблем с использованием pandas:

import pandas as pd

v = [[1, 2, 3],
     [4, 5, 6],
     [7, 8, 9]]

pd.DataFrame(v).rolling(3,axis=0, min_periods=1).mean().values

Вывод:

array([[1. , 2. , 3. ],
       [2.5, 3.5, 4.5],
       [4. , 5. , 6. ]])