python - NumPy 的 : Calculating the average of values between two indices

Question

我需要计算两个指数之间的平均值。假设我的索引是 3 和 10，我想将它们之间的所有值相加并除以值的数量。

最简单的方法就是使用 for 循环，从 3 开始，直到 10，对它们求和，然后除法。这似乎是一种真正的非 pythonic 方式，考虑到 Numpy 提供的功能，我想也许有更短的方式使用一些 Numpy 魔法。非常感谢任何建议

Answer 1

要访问两个索引 i 和 j 之间的所有元素，您可以使用切片:

slice_of_array = array[i: j+1] # use j if you DO NOT want index j included

平均值是用np.average计算的，但在你的情况下你想用元素的数量来加权，所以你可以只使用 np.mean :

import numpy as np
mean_of_slice = np.mean(slice_of_array)

或一次性完成(使用您的索引):

i = 3
j = 10
np.mean(array[i: j+1])