python - 查找与先前元素的元素的平均值并将其求和并平均

Question

我有一组数字，只是为了让它更容易：

import numpy as np
A = np.array([0, 1, 2, 3, 4, 5, ...])

我想要：

((0+1)/2 + (2+1)/2)/2 , ((1+2)/2 + (3+2)/2)/2, ....

我不知道该怎么做。到目前为止我得到了什么：

B = (A[1:]+A[:-1])/2.0

我希望得到：

B = [1, 2, 3, 4, 5 ...]

任何帮助都会很棒。谢谢你。

标签： pythonmath

Answer 1

>> import numpy as np
>> def pair_avg(a):
>>    return (a[1:]+a[:-1])/2.0
>>
>> a = np.arange(10)
>> pair_avg(pair_avg(a))
array([1., 2., 3., 4., 5., 6., 7., 8.])

pair_avg 对 a 中的每一对进行运行平均。据我了解，您想这样做两次。我认为您应该将此操作视为同一函数的递归应用。如果你总是想递归两次，你可以这样做

>> def oper(a):
>>    b = (a[1:]+a[:-1])/2.0
>>    return (b[1:]+b[:-1])/2.0
>>
>> oper(a)
array([1., 2., 3., 4., 5., 6., 7., 8.])

当然，这相当于原始数组元素的某种加权和。当您将其应用于整数时，每 2nd 应用程序pair_avg只会剪掉原始数组的另外两个端点值。