python - Numpy 将编号从列切换到行

Question

我需要更改矩阵的编号方案。说，

import numpy as np
a = np.arange(6).reshape(3,2)
array([[0, 1],
       [2, 3],
       [4, 5]])

我想把它切换到

b = np.array([[0,3],[1,4],[2,5]])
array([[0, 3],
       [1, 4],
       [2, 5]])

所以基本上我首先通过行对矩阵进行编号。我确信在 numpy 中有一个很好的方法可以做到这一点

Answer 1

>>> import numpy as np
>>> np.arange(6).reshape(3,2, order = 'F')
>>> array([[0, 3],
   [1, 4],
   [2, 5]])

来自文档:

http://docs.scipy.org/doc/numpy/reference/generated/numpy.reshape.html

使用 order='F' 指定 Fortran 传统表示，使用 order='C'(默认值)使用传统 C 表示。