pythonowy sposobem korzystania z drugiego warunku w listowych

Question

Załóżmy następującą funkcję:

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 

Jak widać, istnieją dwa różne scenariusze w zależności od par1. Nie podoba mi się to, że linia 3 i linia 5 są prawie identyczne i nie przestrzegają zasady DRY (Do not Repeat Yourself). W jaki sposób można poprawić ten kod?

Answer 1

Może to działa:

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

tak przyzwyczajeni tak:

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

Można wybrać funkcję warunek dynamicznie przy użyciu funkcji, która po prostu zwraca True w pierwszym przypadku i który faktycznie porównuje wynik my_fun2 z par1 w drugim przypadku:

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)] 

lub przez zastąpienie zewnętrzną pętlę z generator expression w przypadku par1 nie jest pusty ciąg znaków:

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] 

Jednak nie pozwól DRY uczynić funkcja trudniejsze do odczytania, utrzymać lub debug. Osobiście uważam, że twoje podejście jest w porządku (i prawdopodobnie szybciej) i prawdopodobnie nie powinieneś niczego zmieniać.

Answer 3

dlaczego nie używać filtra?

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)