Если еще в понимании списка

У меня есть список l:

l = [22, 13, 45, 50, 98, 69, 43, 44, 1]

Для номеров выше 45 включительно, я хотел бы добавить 1; и для чисел меньше его 5.

Я старался

[x+1 for x in l if x >= 45 else x+5]

Но это дает мне синтаксическую ошибку. Как я могу достичь if - else как это в понимании списка?

8 ответов

Решение
>>> l = [22, 13, 45, 50, 98, 69, 43, 44, 1]
>>> [x+1 if x >= 45 else x+5 for x in l]
[27, 18, 46, 51, 99, 70, 48, 49, 6]

Сделай что-нибудь, если <condition>, еще делай - что-то еще.

Причина, по которой вы получаете эту ошибку, связана с тем, как выполняется понимание списка.

Имейте в виду следующее:

[ expression for item in list if conditional ]

Эквивалентно:

for item in list:
    if conditional:
        expression

Где expression в немного другом формате (подумайте о переключении темы и порядка глаголов в предложении).

Поэтому твой код [x+1 for x in l if x >= 45] Является ли это:

for x in l:
    if x >= 45:
        x+1

Тем не менее, этот код [x+1 if x >= 45 else x+5 for x in l] делает это (после перестановки expression):

for x in l:
    if x>=45: x+1
    else: x+5
[x+1 if x >= 45 else x+5 for x in l]

И для вознаграждения, вот комментарий, который я написал, чтобы запомнить это в первый раз, когда я сделал эту ошибку:

Условное выражение Python a if C else b и не может быть использован как:

[a for i in items if C else b]

Правильная форма:

[a if C else b for i in items]

Даже если есть действительная форма:

[a for i in items if C]

Но это не то же самое, что фильтровать по C, но они могут быть объединены:

[a if tC else b for i in items if fC]

Вы должны поместить выражение в начало понимания списка, оператор if в конце фильтрует элементы!

[x+1 if x >= 45 else x+5 for x in l]

Как в [a if condition1 else b for i in list1 if condition2], два ifс condition1 а также condition2 делать две разные вещи. Часть (a if condition1 else b) из лямбда-выражения:

lambda x: a if condition1 else b

в то время как другой condition2 это еще одна лямбда

lambda x: condition2

Понимание всего списка можно рассматривать как комбинацию map а также filter:

map(lambda x: a if condition1 else b, filter(lambda x: condition2, list1))

Вы также можете поместить условное выражение в скобки внутри списка понимания:

    l = [22, 13, 45, 50, 98, 69, 43, 44, 1]
    print [[x+5,x+1][x >= 45] for x in l]

[false, true] [условие] - это синтаксис

У меня просто была похожая проблема, и я нашел этот вопрос и ответы действительно полезными. Вот часть, в которой я был смущен. Я пишу это явно, потому что никто не сказал это просто на английском языке:

Итерация идет в конце.

Обычно цикл идет

for this many times:
    if conditional: 
        do this thing
    else:
        do something else  

Все утверждают часть понимания списка просто, как первый ответ,

[ expression for item in list if conditional ] 

но это на самом деле не то, что вы делаете в этом случае. (Я пытался сделать это таким образом)

В этом случае это больше похоже на это:

[ expression if conditional else other thing for this many times ] 

Вы можете переместить условное в:

v = [22, 13, 45, 50, 98, 69, 43, 44, 1]
[ (x+1 if x >=45 else x+5)  for x in v ]

Но это начинает выглядеть немного некрасиво, так что вам может быть лучше использовать обычный цикл. Обратите внимание, что я использовал v вместо l для переменной списка, чтобы уменьшить путаницу с номером 1 (я думаю, l а также O следует избегать использования в качестве имен переменных при любых обстоятельствах, даже в быстром и грязном примере кода).

Другие вопросы по тегам