Transpose of a Matrix

Write a Python function that computes the transpose of a given matrix.

Example:

Input:

a = [[1, 2, 3],[4, 5, 6]]

Output:

[[1, 4],[2, 5],[3, 6]]

Reasoning:

The transpose of a matrix is obtained by flipping rows and columns.

Code:

import numpy as np

def transpose_matrix(a):
    return np.array(a).T