一維數組的Numpy Element-wise Mean

Question

我想計算一維數組中元素的按元素均值。

>>> a = np.array([1, 3, 5, 7])
>>> b = element_wise_mean(a)
>>> b
array([2., 4., 6.])

除了簡單的自定義功能以外，還有什么可以做的了嗎？

Answer 1

使用以下代碼：

>>> (a[:-1]+a[1:])/2
array([ 2.,  4.,  6.])

采取以下步驟：

>>> a[:-1]
array([1, 3, 5])
>>> a[1:]
array([3, 5, 7])
>>> a[:-1]+a[1:]
array([ 4,  8, 12])
>>> (a[:-1]+a[1:])/2
array([ 2.,  4.,  6.])

一種更通用的方法是在N元素上使用移動平均濾波器（代碼取自青金石 ，Paul Panzer則添加了其他內容）。 在您的情況下，它將平均兩個元素：

>>> N=2
>>> np.convolve(a, np.ones((N,))/N, mode='valid')
array([ 2.,  4.,  6.])

>>> N=3
>>> np.convolve(a, np.ones((N,))/N, mode='valid')
array([ 3.,  5.])