python - 使用 numpy.mean 平均直到数字 N

Question

假设我有一个包含 10 个元素的列表 a

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

我只想取 N 元素的平均值，比如说 4，也就是说，[1,2,3,4] 的平均值 我如何使用 Numpy 做到这一点？

通常我可以遍历一个列表，但这会花费我一个额外的列表，我不想invest。

Numpy文档没有为我提供任何东西。

Answer 1

假设您有一个 numpy 数组而不是列表，只需将前 N 元素切片并使用 mean 方法。

(如果你想让事情正常工作，不管它是列表还是 numpy 数组，请使用 np.mean(a[:4])，正如@larsmans 上面所建议的。)

例如

import numpy as np
x = np.arange(10)
print x
print 'Mean of first 4 items in x', x[:4].mean()

切片 numpy 数组即 y = x[:4] 不会使用 numpy 数组进行复制(就像使用列表一样)。相反，它将“ View ”返回到 x 中。这不会复制内存，因此非常高效。

对部分 numpy 数组进行切片和操作是一种非常常见的习惯用法。查看文档以获取更多信息:http://docs.scipy.org/doc/numpy/reference/arrays.indexing.html