python - 是否有 numpy.average() 的替代方法可以在使用 numpy 时沿轴平均？

Question

我试图通过使用 pypy 使一些代码运行得更快。但是，每当我使用 pypy 运行代码时，我都会从 pypy 的 numpy 等价物(numpy ).

在我通常的代码中，我使用...

numpy.average(array, axis=0)

为了沿特定轴对数组进行平均。

当使用 pypy 和 numpypy 运行相同的代码时，出现以下错误:

TypeError: average() got an unexpected keyword argument 'axis'

我可以使用 for 循环遍历数组并对单个元素进行平均，但这会花费很长时间，而且(我想)无法提供我想要的速度。

是否有 numpy.average() 的替代方法可以在使用 numpypy 时沿轴平均？

Answer 1

如果你不需要使用weights参数，你可以使用mean代替(1.9.1-dev0 -- 不知道什么时候引入的):

>>>> import numpypy as np
>>>> a = np.arange(2*3).reshape(2,3)
>>>> a
array([[0, 1, 2],
       [3, 4, 5]])
>>>> np.mean(a, axis=0)
array([ 1.5,  2.5,  3.5])
>>>> np.mean(a, axis=1)
array([ 1.,  4.])
>>>> a.mean(axis=0)
array([ 1.5,  2.5,  3.5])
>>>> a.mean(axis=1)
array([ 1.,  4.])