façon Pythonic d'utiliser la deuxième condition dans la liste compréhensions

Question

Supposons que la fonction suivante:

def myfun(my_list, n, par1=''): 
    if par1 == '': 
     new_list = [[my_fun2(i,j) for j in range(n)] for i in range(n)] 
    else: 
     new_list = [[my_fun2(i,j) for j in range(n)] for i in range(n) if my_fun2(i,n) == par1] 
    return new_list 

Comme vous pouvez le voir, il y a deux scénarios différents selon par1. Je n'aime pas que la ligne 3 et la ligne 5 soient presque identiques et ne suivent pas le principe DRY (Do not Repeat Yourself). Comment ce code peut-il être amélioré?

Answer 1

Cela pourrait fonctionner:

new_list = [[my_fun2(i,j) for j in range(n)] for i in range(n) if par1 == '' or my_fun2(i,n) == par1] 

donc utilisé comme ceci:

def myfun(my_list, n, par1=''): 
    return [ 
       [my_fun2(i,j) for j in range(n)] 
       for i in range(n) if par1 == '' or my_fun2(i,n) == par1 
      ] 

Answer 2

Vous pouvez choisir la fonction de condition dynamique à l'aide d'une fonction qui retourne juste True dans le premier cas et qui fait compare le résultat my_fun2 avec par1 dans le second cas:

def myfun(my_list, n, par1=''): 
    if par1 == '': 
     cond = lambda x, y: True 
    else: 
     cond = lambda i, n: my_fun2(i, n) == par1 
    return [[my_fun2(i,j) for j in range(n)] for i in range(n) if cond(i,n)] 

ou en remplaçant la boucle extérieure avec un generator expression en cas par1 n'est pas une chaîne vide:

def myfun(my_list, n, par1=''): 
    if par1 == '': 
     outer = range(n) 
    else: 
     # a conditional generator expression 
     outer = (i for i in range(n) if my_fun2(i,n) == par1) 
    return [[my_fun2(i,j) for j in range(n)] for i in outer] 

Cependant ne font pas laisser sécher la fonction plus difficile à lire, maintenir ou débogage. Personnellement, je pense que votre approche est bonne (et probablement plus rapide) et vous ne devriez probablement rien changer.

Answer 3

pourquoi ne pas utiliser un filtre?

from operator import eq 
def myfun(my_list, n, par1=''): 
    new_list = ([my_fun2(i,j) for j in range(n)] for i in range(n)) 
    if par1 != '': 
     new_list = filter(eq(par1),new_list) 
    return list(new_list)