nanの入れ替えをしたい

Question

python3で、配列内に計算上発生したnanを0で入れ替えるプログラムを書きたいと思っているのですが、入れ替えができません。

    for i in range(0,10):
        for j in range(0,10):
            if G[i,j] == np.nan:
                G[i,j] = 0
                print(G[i,j])

replaceも試したのですが同じ結果となってしまいました。
どなたか御教授いただければ嬉しいです。

Answer 1

== np.nanではなくnumpy.isnanで判定してください。
さらにnumpy.whereやnumpy.nan_to_numを使えば一括置換できます。

import numpy as np
G = np.array([[1,2,np.nan],[4,np.nan,6]])
print(G)

# numpy.whereで一括置換
G2 = np.where( np.isnan(G),0,G)
print(G2)

# numpy.nan_to_numで一括置換
G3 = np.nan_to_num(G)
print(G3)

for i in range(G.shape[0]):
    for j in range(G.shape[1]):
        if np.isnan(G[i,j]):
        #if G[i,j] == np.nan:
            print('nan!')
            G[i,j] = 0
        print(G[i,j])
print(G)

Comments

[w_xox_w]

なるほど、numpyにそのような関数があるのですね、勉強になります。
ありがとうございます！