Matrix Transposition in Python