if...else em Python List Comprehension

Este tutorial irá explicar várias maneiras de realizar a compreensão de lista com a instrução if ... else em Python. A compreensão de uma lista é uma maneira curta e sintática de criar uma lista com base em uma lista existente. Geralmente é usado para criar uma nova lista filtrada ou alterada de uma lista atual.

Por exemplo, temos uma lista python ['Ali','Mark', None, 'Sara', None, 'Rahul'] e queremos criar uma nova lista ['Ali','Mark', '', 'Sara', '', 'Rahul'], podemos fazer isso usando a compreensão de lista.

Python if ... else Compreensão da lista

O código de exemplo a seguir demonstra como podemos criar uma lista alterada a partir da lista existente usando a compreensão de lista com a instrução if ... else:

my_list = ["Ali", "Mark", None, "Sara", None, "Rahul"]

new_list = [str(x.strip()) if x is not None else "" for x in my_list]
print(new_list)

Resultado:

['Ali', 'Mark', '', 'Sara', '', 'Rahul']

A sintaxe geral de compreensão de lista em Python com if ... else é:

[f(x) if condition else g(x) for x in list]

Se condition for verdadeira para o elemento da lista x, f(x), qualquer função aplicável, é aplicada ao elemento; caso contrário, g(x) será aplicado.

Código de exemplo:

my_list = ["Ali", "Mark", None, "Sara", None, "Rahul"]

new_list = [x.upper() if x is not None else "" for x in my_list]
print(new_list)

Resultado:

['ALI', 'MARK', '', 'SARA', '', 'RAHUL']