Python преобразование одномерного массива в двумерный массив с помощью «строк» ​​или «столбцов»

Question

Я хотел бы контролировать, как array.reshape () заполняет новый массив. Например,

a = np.arange(12).reshape(3,4)
## array([[ 0,  1,  2,  3],
##   [ 4,  5,  6,  7],
##   [ 8,  9, 10, 11]]) 

, но я хотел бы иметь возможность заполнить массив по столбцам чем-то вроде:

a = np.arange(9).reshape(3,3, 'columnwise')
## array([[ 0,  3,  6,  9],
##   [ 1,  4,  7,  10],
##   [ 2,  5, 8, 11]]) 

Answer 1

Использование np.transpose.

import numpy as np

print(np.arange(9).reshape(3,3).transpose())

Вывод:

[[0 3 6]
 [1 4 7]
 [2 5 8]]

Answer 2

In [22]:  np.arange(12).reshape(3,4, order='F')                                                        
Out[22]: 
array([[ 0,  3,  6,  9],
       [ 1,  4,  7, 10],
       [ 2,  5,  8, 11]])

Answer 3

Если вы сделаете транспонирование оригинальной матрицы, вы получите желаемый эффект.

import numpy as np
a = np.arange(6).reshape(3,3).tranpose()

ИЛИ

a = np.arange(6).reshape(3,3).T