numpy.where: Wie verzögert man die Auswertung von Parametern?

Question

Ich verwende numpy.where, und ich fragte mich, ob es einen einfachen Weg gab, den unbenutzten Parameter zu vermeiden. Beispiel:

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

kehrt:

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

, aber ich erhalte eine Division durch Null Warnung, weil, wenn z 0 war, noch der Code 1/z auswertet, obwohl es sie nicht verwendet werden.

Answer 1

Sie eine Maske anwenden kann:

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

Answer 2

Sie können auch die Warnung ausschalten und wieder einschalten, nachdem Sie über den Kontextmanager errstate getan werden:

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

Answer 3

scipy hat eine wenig Dienstprogramm nur für diesen Zweck: https://github.com/scipy/scipy/blob/master/scipy/_lib/_util.py#L26

Es ist im Grunde eine Sache, die aus der Zuteilung Setzen Sie Array und verwenden Sie np.place (oder Putmask), um genau das zu füllen, was Sie brauchen.