find transpose of a matrix in python