0除算を回避したい！

Question

前提・実現したいこと

0除算が起きた時に0を出力できるようにしたい！
具体的には,X_train/Y_train=0になるようにしたいです！

該当のソースコード

data = np.arctan((X_train/Y_train))

試したこと

if文などを駆使しましたがうまくいきませんでした

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

X_trainとY_trainはnp.arrayでデータを取ってきてます！
また,jupyter notebookで書いてます,よろしくお願いいたします。

Answer 1

np.arctan2を使えばよいのでは?

data = np.arctan2(X_train, Y_train)

Comments

[mahc]

そんなのがあるんですね...
大変勉強になりました,ありがとうございます！

Answer 2

そもそも、numpy の場合 0除算のデフォルトの挙動は'Warning'ですので、エラーにはならないと思うのですが。。

'Warning'表示すら表示したくない場合は numpyの0除算時の挙動を 'ignore' に変更するとよいかと思います。

np.seterr(divide='ignore')

で、0除算が行われた際の計算結果は

• 正の値 / 0 の結果は np.inf
• 負の値 / 0 の結果は -np.inf
• 0 / 0 の結果は np.nan

となりますので、

具体的には,X_train/Y_train=0になるようにしたいです！

なのであれば、結果が np.inf, -np.inf, np.nan の値を 0 に置き換えたらよいのではないでしょうか。

import numpy as np

np.seterr(divide='ignore', invalid='ignore')
X = np.array([4,4,-4,0,0])
Y = np.array([2,0,0,0,1])
result = X/Y
#[  2.  inf -inf  nan   0.]
result = np.where(np.isinf(result) | np.isnan(result), 0, result)
#[2,0,0,0,0]

Comments

[mahc]

なるほど,勉強になります！
大変参考になりました！

Answer 3

割り算する前に０かどうかを判断すればいいのでは

Comments

[mahc]

今後,バッチ処理をするので,できればコード上でどうにかしたいです！

[y_waiwai]

コード上ですればいいじゃない
まあ、実用的にするなら、割り算関数を作って、第２引数が０なら０を返すようにすればいいだけでしょ
data = np.arctan(divz(X_train,Y_train))

[mahc]

なるほど...
ありがとうございます！
大変勉強になりました.