Transpose a matrix (2D array) in Python

transpose.py

def transpose(matrix):
    return [[row[i] for row in matrix] for i in range(len(matrix[0]))]


# this one is a bit slower
def t2(matrix):
    return map(list, zip(*matrix))