Wiele warunków IF w zrozumieniu listy Pythona

Question

Zastanawiałem się, czy jest możliwe umieszczenie wielu warunków if w zrozumieniu listy? Nie znalazłem czegoś takiego w dokumentach.

Chcę być w stanie zrobić coś takiego

ar=[] 
for i in range(1,n): 
    if i%4 == 0: ar.append('four') 
    elif i%6 == 0: ar.append('six') 
    else: ar.append(i) 

podstawie listy ze zrozumieniem. Jak mogę to zrobić?

Czy to możliwe? Jeśli nie, jaki byłby najbardziej elegancki (pythonic) sposób, aby to osiągnąć?

Answer 1

Jak o

ar = [('four' if i % 4 == 0 else ('six' if i % 6 == 0 else i)) for i in range(1, n)] 

Na przykład, jeśli n = 30 to

[1, 2, 3, 'four', 5, 'six', 7, 'four', 9, 10, 11, 'four', 13, 14, 15, 'four', 17, 'six', 19, 'four', 21, 22, 23, 'four', 25, 26, 27, 'four', 29] 

ETA: Oto w jaki sposób można zastosować listę warunków:

CONDITIONS = [(lambda i: i % 4 == 0, "four"), (lambda i: i % 6 == 0, "six"), 
       (lambda i: i % 7 == 0, "seven")] 

def apply_conditions(i): 
    for condition, replacement in CONDITIONS: 
     if condition(i): 
      return replacement 
    return i 

ar = map(apply_conditions, range(0, n)) 

Answer 2

można umieścić cię logiki w oddzielnej funkcji, a następnie mają elegancję rozumienia listy razem z nią h czytelność funkcji:

def cond(i): 
    if i % 4 == 0: return 'four' 
    elif i % 6 == 0: return 'six' 

    return i 

l=[cond(i) for i in range(1,n)]