nanをその配列の平均値にしたい

前提・実現したいこと

配列のnanをその配列の平均値にしたい。

発生している問題・エラーメッセージ

エラーは出ていない。

該当のソースコード

data = [[3. 6. 1. 1. 2. 2. 1. 7.]
 [2. 1. 1. 7. 6. 1. nan 1.]
 [4. 3. 5. 3. 1. 2. 4. 7.]]

とdata変数にnanを要素にもった二次元配列が入っている。このnanの要素をこの配列の平均値に入れ替えたい。例えば、[2. 1. 1. 7. 6. 1. nan 1.]のnanを[2. 1. 1. 7. 6. 1. nan 1.]の配列のnan以外の要素の平均値((2+1+1+7+6+1+1)/7)である2.7で置換したい。

試したこと

https://note.nkmk.me/python-pandas-nan-dropna-fillna/
などの記事を参考にして、置換ができないか考えたが、
どれも配列の要素の番号を指定して置換しているもので、目的と違った。
自分は、要素でnanがある配列は、自動的にその配列の要素の平均値で置換するシステムを作りたい。

そしてその方法を知りたい。

補足情報（FW/ツールのバージョンなど）

Pythonのデータ分析ライブラリを使い、それを実現したい。

行動規範の内容に同意します

回答2件

ベストアンサー

python
1>>> data = np.array([[3., 6., 1., 1., 2., 2., 1., 7.], [2., 1., 1., 7., 6., 1., np.nan, 1.], [4., 3., 5., 3., 1., 2., 4., 7.]])
2>>> mean = np.nanmean(data)
3>>> data[np.isnan(data)] = mean
4>>> data
5array([[3.        , 6.        , 1.        , 1.        , 2.        ,
6        2.        , 1.        , 7.        ],
7       [2.        , 1.        , 1.        , 7.        , 6.        ,
8        1.        , 3.08695652, 1.        ],
9       [4.        , 3.        , 5.        , 3.        , 1.        ,
10        2.        , 4.        , 7.        ]])