python - numpy配列の複数軸の平均を取得する

Question

numpy では、複数の軸の平均を計算する高速な方法はありますか? n 次元配列の 0 軸以外のすべての平均を計算しています。

私は現在これを行っています。

for i in range(d.ndim - 1):
    d = d.mean(axis=1)

Pythonループを使用しないソリューションがあるかどうか疑問に思っています。

Answer 1

私のアプローチは、配列を再形成して、より高い次元をすべて平坦化し、軸 1 で平均を実行することです。これはあなたが探しているものですか?

In [14]: x = np.array([[[1,2],[3,4]],[[5,6],[7,8]]])

In [16]: x.reshape((x.shape[0], -1)).mean(axis=1)
Out[16]: array([ 2.5,  6.5])

(ステップ 2 は、より高い次元の長さの積を計算するだけです)

Answer 2

@dsg101 の提案に続いて、これはあなたが望むようなものですか?

>>> import numpy as np
>>> d=np.reshape(np.arange(5*4*3),[5,4,3])
>>> d
array([[[ 0,  1,  2],
        [ 3,  4,  5],
        [ 6,  7,  8],
        [ 9, 10, 11]],

       [[12, 13, 14],
        [15, 16, 17],
        [18, 19, 20],
        [21, 22, 23]],

       [[24, 25, 26],
        [27, 28, 29],
        [30, 31, 32],
        [33, 34, 35]],

       [[36, 37, 38],
        [39, 40, 41],
        [42, 43, 44],
        [45, 46, 47]],

       [[48, 49, 50],
        [51, 52, 53],
        [54, 55, 56],
        [57, 58, 59]]])
>>> np.mean(np.reshape(d,[d.shape[0],np.product(d.shape[1:])]),axis=1)
array([  5.5,  17.5,  29.5,  41.5,  53.5])