NUMPY061 题解：计算累积平均

题目

给定数组，计算累积平均值。

解题思路

1. 计算累积和
2. 除以对应的元素个数得到累积平均

代码

import numpy as np

n = int(input())
arr = np.array(list(map(float, input().split())))
cumsum = np.cumsum(arr)
result = cumsum / np.arange(1, n + 1)
print(result)

代码详解

• $\texttt{np.cumsum}$ —— 计算累积和
• $\texttt{np.arange(1, n + 1)}$ —— 生成 [1, 2, 3, ..., n]
• 相除得到累积平均