return i for i if statement in list pyhton code example

Example: 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)