【python】0除算エラー回避の仕方

Question

ψ1,ψ2にK1,K2を代入した際に（エラー文）divide by zero encountered in true_divide
がでてしまうのでzの中にあるz=0を除いてzを代入し，K1,K2およびψ1,ψ2を計算したいです．
K1,K2(0<z)

また，z_depth=5のとき，z=0を除けば，zの要素数len(z)=50,yの要素数len(y)=51，となり，y,zの要素数が変わってしまうのですが，ψに代入できますか？

import math
import numpy as np

##########
π = math.pi
Pn = 2200
z_depth = 5
c = 0.5
z = np.arange(0,(z_depth+0.1)*10)/10
y_c = np.arange((-z_depth*5),(z_depth+0.1)*5)/10
y = y_c*c
##########
yv, zv = np.meshgrid(y, z)

K1 = ((c+yv)**2)+zv**2 #(z<0)の範囲でK1,K2を決定
K2 = ((c-yv)**2)+zv**2 #z=0,y=-c,+cのとき，K1,K2が0となるため
    
#ψ1 = π*(1-((K2/K1)**(0.5)))/K1*((K2/K1)**(0.5))*((2*(K2/K1)**(0.5))+(K1+K2-4*(c**2))/K1)**(0.5)

#ψ2 = π*(1+((K2/K1)**(0.5)))/K1*((K2/K1)**(0.5))*((2*(K2/K1)**(0.5))+(K1+K2-4*(c**2))/K1)**(0.5)

input()

Answer 1

numpy の場合 0除算のデフォルトの挙動はErrorではなくて単なるWarningだった気がしますが、本当にエラー（例外）が発生していますか？

とりあえず、0除算時の挙動を

print(np.geterr()['divide'])

にて確認してみてください。
この結果が 'worm'（ワーニング）や**'print'（ワーニングを標準出力）、'log'(ワーニングをログ出力)'ignore'（無視）の場合はエラーにはならないはずです。
もし、結果が、''raise**（例外）などになっている場合は

# ワーニングを表示するが処理は続行する
np.seterr(divide='warn')
```Python```
または
```Python
# ワーニングも出さない
np.seterr(divide='ignore')

を行い、0除算時の挙動を変更すると良いかと思います。

https://docs.scipy.org/doc/numpy/reference/generated/numpy.seterr.html
https://docs.scipy.org/doc/numpy/reference/generated/numpy.geterr.html

で、上記の設定を行った際の0除算時の結果ですが、

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

として扱われます。

Answer 2

• そもそも０除算にならないようにコードを考え直す
• 除算の前に０かどうかを判断して処理をする
• ０除算の例外を補足するようにして後でどうにかする

と、こんなもんでしょうか