python數組中的移動平均線

Question

我有一個形狀等於（1000,151）的數組'aN'。 我需要計算每行10個數據的平均值，所以我實現了這個

arr = aN[:]
window_size = 10
i = 0
moving_averages = []

while i < len(arr) - window_size + 1:
    window_average = round(np.sum(arr[i:i+window_size]) / window_size, 2)      
    moving_averages.append(window_average)
    i += 10

關鍵是我的輸出是一個包含 100 個數據的列表，但我需要一個與原始數組 (151) 具有相同列數的數組。 關於如何得到這個結果的任何想法？

蒂亞！！

Answer 1

如果將其轉換為 pandas 數據幀，則可以將 pandas 的rolling()函數與mean()函數一起使用。 它應該能夠完成你所需要的。