Find transpose of matrix using python