python list omprehension if and code example
Example 1: if else in list comprehension
[f(x) if condition else g(x) for x in sequence]
Example 2: list comprehension for loop and if
matrix = [[1, 2], [3,4], [5,6], [7,8]]
transpose = [[row[i] for row in matrix] for i in range(2)]
print (transpose)