Skip to main content
 首页 » 编程设计

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

2025年05月04日15开发

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

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 中有一个很好的方法可以做到这一点

请您参考如下方法:

>>> 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 表示。