2次元配列のnp.nanmeanの計算方法

Question

2次元配列を含む辞書があります。私はこの方法で平均を計算しようとしましたが、配列にもnan値が含まれているため動作しません。平均を計算する簡単な方法はありますか？

All = np.zeros(385000).reshape(550,700) 

for i in dic.keys(): 
    a = dic[i]['data'] 
    avg = (All+a)/len(dic.keys()) 

Answer 1

あなたがNaNsを無視して、両方の入力aとb渡っ要素ごと考慮平均を発見しようとしているようです。したがって、1つの方法は、np.dstackでこれらの2つの配列を積み重ねることです。aとbを3番目の軸に沿って積み重ねて、同じ軸に沿ってnp.nanmeanを単純に使用します。

np.nanmean(np.dstack((a,b)),axis=2) 

サンプル実行 - - 掲載のコードに示すように、あなたが辞書からのもの2Dの配列を取得することとなる場合については

In [28]: a 
Out[28]: 
array([[ 2., nan], 
     [ 5., 4.]]) 

In [29]: b 
Out[29]: 
array([[ nan, 3.], 
     [ 7., 2.]]) 

In [30]: np.nanmean(np.dstack((a,b)),axis=2) 
Out[30]: 
array([[ 2., 3.], 
     [ 6., 3.]]) 

したがって、我々はそうのようなシンプル実装を持っているでしょうループの理解度を使って、配列をnp.dstackの3D配列として収集し、最終的にはnp.nanmeanを最後の軸に沿って使用すると、

のようになります0

np.nanmean(np.dstack([d['data'] for d in dic]),axis=2)