python - 每 2 个 Numpy 元素的加权平均值

Question

假设我有这个数组:[4,3,2]我想获得每两个元素的加权平均值。因此，我会得到一个如下所示的数组(假设权重为 [0.6, 0.4, 0.6]:[3.6、2.4]

知道如何使用大型数组(5000)做到这一点吗？

非常感谢!

Answer 1

您可以查看numpy roll() 方法。

import numpy as np
weight = 0.5
x = np.array([4, 3, 2])
(x*weight + np.roll(x, -1)*(1-weight))[:-1]

输出:

array([ 3.5,  2.5])

Roll 方法将移动原始数组，将移动后的数组与原始数组相加，然后将每个数组乘以各自的权重(权重总和为 1)，您将得到加权平均值。所有这些都是矢量化操作，这将使执行非常顺利。

您可以调整权重以使第一个或第二个元素产生更大的影响；