numpy.where：評価パラメータを遅らせる方法は？

Question

私はnumpy.whereを使用していますが、使用されていないパラメータを呼び出さないようにする簡単な方法があるかどうかは疑問でした。例：

import numpy as np 
z = np.array([-2, -1, 0, 1, -2]) 
np.where(z!=0, 1/z, 1) 

リターン：

array([-0.5, -1. , 1. , 1. , -0.5]) 

しかしzが0だったとき、それはそれを使用していなくても、コードはまだ1/zを評価するので、私はゼロ警告による除算を取得します。

あなたがマスクを適用することができます

Answer 1

：

out = numpy.ones_like(z) 
mask = z != 0 
out[mask] = 1/z[mask] 

Answer 2

ます。また、警告をオフにして、コンテキストマネージャerrstate使用 を行われた後、オンに戻すことができます：

with np.errstate(divide='ignore'): 
    np.where(z!=0, 1/z, 1) 

Answer 3

scipyのダウンロードが持っているのこの目的のための小さなユーティリティ： https://github.com/scipy/scipy/blob/master/scipy/_lib/_util.py#L26

これは基本的に配列を入れ、np.place（またはputmask）を使って必要なものを正確に埋めてください。