Valeur par défaut pour argument optionnel en Python:

Question

je la méthode suivante:

def get_data(replace_nan=False): 
    if replace_nan is not False 
     data[numpy.isnan(data)] = replace_nan 
     return data 
    else: 
     return data[~numpy.isnan(data)] 

Donc, si replace_nan est faux, nous revenons une série de données, mais supprimer NaN s, et si elle est autre chose, nous remplaçons NaN s avec l'argument. Le problème est, je peux vouloir remplacer NaN avec False. Ou toute autre chose, pour ça. Quelle est la manière la plus pythonique de le faire? Ce:

def get_data(**kwargs): 
    if "replace_nan" in kwargs: 
     ... 

fonctionne, mais est sémantiquement laid (parce que nous sommes intéressés vraiment juste un argument de mot-clé, replace_nan) Toute suggestion comment gérer ce cas?

Answer 1

Habituellement, les gens utilisent None comme valeur par défaut, puis vérifient is not None.

Si vous devez autoriser None, aussi, utiliser un objet factice:

__default = object() 
def get_data(replace_nan=__default): 
    if replace_nan is __default: 
     ... 

Answer 2

numpy évalue tableau Faux l'intérieur à 0:

>>>np.array([False,True,2,3]) 
    array([0, 1, 2, 3]) 

Donc, cela pourrait probablement pas ce que vous voulez arriver .

def get_data(replace_nan=False): 
     if replace_nan: 
      return np.where(np.isnan(data),replace_nan,data) 
     else: 
      return data[~numpy.isnan(data)] 

La fonction numpy.where crée un tableau avec les index où vos entrées sont NaN. Là, il remplace les entrées avec replace_nan, partout ailleurs il conserve les entrées.

De l'manual page:

numpy.where(condition[, x, y]) 
Return elements, either from x or y, depending on condition. 

Answer 3

Je voulais mettre cela comme commentaire ci-dessous la réponse de ThiefMaster mais pas de formatage dans les commentaires autorisés, alors ...:

Si vous êtes préoccupé par encombrer votre espace de noms que vous peut-avec quelques astuces- del la variable après avoir défini la fonction.

__default = object() 
def get_data(replace_nan=__default, __default=__default): 
    if replace_nan is __default: 
    ... 
del __default 

Ou:

__default = object() 
def get_data(replace_nan=__default): 
    if replace_nan is get_data.default_replace_nan: 
    ... 
get_data.default_replace_nan = __default 
del __default 

Answer 4

Une autre façon d'éviter l'encombrement de l'approche de ThiefMaster est la suivante:

def get_data(replace_nan=object()): 
    if replace_nan is get_data.func_defaults[0]: 
    ... 

Mais il utilise python interna qui pourrait ne pas être aussi portable (pypy/stackles/prochaine version/...).