transpose matrix in python