python if elif else list comprehension

1>>> l = [1, 2, 3, 4, 5]
2>>> ['yes' if v == 1 else 'no' if v == 2 else 'idle' for v in l]
3['yes', 'no', 'idle', 'idle', 'idle']
4

1# Basic syntax for if:
2desired_elements = [f(x) for x in iterable_object if condition]
3
4# Basic syntax for if else:
5desired_elements = [f(x) if condition else g(x) for x in iterable_object]
6
7# Basic syntax for if elif else:
8desired_elements = [f(x) if condition elif condition_2 g(x) else h(x) for x in iterable_object]
9
10# Example usage:
11# Say you want to return the square of elements if elements are less 
12# than 10, otherwise you just want to return the elements themselves:
13your_list = [1, 7, 13, 11, 23, 2, 17, 42, 8, 5]
14new_list = [x**2 if x < 10 else x for x in your_list]
15print(new_list)
16--> [1, 49, 13, 11, 23, 4, 17, 42, 64, 25]

1matrix = [[1, 2], [3,4], [5,6], [7,8]]
2transpose = [[row[i] for row in matrix] for i in range(2)]
3print (transpose)