if in python list comprehension code example
Example 1: python if else list comprehension
desired_elements = [f(x) for x in iterable_object if condition]
desired_elements = [f(x) if condition else g(x) for x in iterable_object]
desired_elements = [f(x) if condition elif condition_2 g(x) else h(x) for x in iterable_object]
your_list = [1, 7, 13, 11, 23, 2, 17, 42, 8, 5]
new_list = [x**2 if x < 10 else x for x in your_list]
print(new_list)
--> [1, 49, 13, 11, 23, 4, 17, 42, 64, 25]
Example 2: list comprehension if
[f(x) for x in sequence if condition]