用于评估庞大数据集平均值的 Pythonic 方法

Question

我有一个向量 X，它由 N = 10e6 值组成。我想计算增加对的平均值。例如:

for i in range(0,N-1):
    Ex[i] = X[0:i+1].mean()

这是一种非常低效的方法。什么是更智能的 Python 算法？注意 Ex 和 X 都是浮点值的 numpy 数组。

Answer 1

以 numpy 为中心的解决方案如下:

X = np.random.rand(10**6)
EX = np.cumsum(X) / np.arange(1, X.shape[0]+1)