numpy.selectでdiv/0エラーを避ける

Question

numpy.selectを使用して配列に条件付きで値を割り当てようとしています。しかし、選択の使用は、DIV/0エラーが発生する可能性がいうだけで、関連する割り当てよりも、すべての可能な割り当て、例えばの計算を必要とします。

import numpy as np 
def testfunc(z): 
    conditionlist = [z < 0, z == 0, z > 0] 
    choicelist = [1/z, 0, 1 + z] 
    return np.select(conditionlist, choicelist) 

if __name__ == "__main__": 

    print testfunc(np.array([0])) 

このコードは、DIV/0エラーで失敗します

、1/Zもののz = 0は、実際に返される配列に代入する必要はありません。

div/0エラーを実行しないでnumpy配列に条件付きで値を割り当てる方法はありますか？ループは唯一のオプションですか？

Answer 1

np.select([z < 0, z == 0, z > 0], [1/(z + (z == 0)), 0, 1 + z]) 

z == 0はブール値の配列を与える。これをzに追加すると、np.selectが使用するインデックスのために、ゼロ要素のない配列がzに等しくなります。

Answer 2

numpy.seterrを使用すると、ゼロ除算の動作をraiseから他のものに変更できます。