Изменение формы Numpy массив

Question

Я хотел бы изменить следующий Numpy массив в IPython:

array([[1, 2, 3, 4],[5, 6, 7, 8],[9, 10, 11, 12]]) # Array A 

к:

array([[1, 5, 9],[2, 6, 10],[3, 7, 11],[4, 8, 12]]) # Array B 

Основная задача состоит в том, чтобы вычислить среднее значение первых элементов массива A (что означает среднее значение 1,5,9), затем два элемента и т. д.

Я думал, что самый простой способ сделать это - изменить матрицу и затем вычислить ее среднее значение.

Есть ли способ сделать это без прохождения через массив через цикл for?

Answer 1

Для выполнения таких расчетов вы должны использовать numpy.

Предполагая, что ваш исходный массив:

a.transpose() 

будет делать трюк

Answer 2

Используйте axis ключевое слово на mean; нет необходимости reshape:

>>> A = np.array([[1, 2, 3, 4],[5, 6, 7, 8],[9, 10, 11, 12]]) 
>>> A.mean(axis=0) 
array([ 5., 6., 7., 8.]) 

Если вы хотите, чтобы массив B аут, то вам не нужно транспонировать массив, а не изменить его форму:

>>> A.T 
array([[ 1, 5, 9], 
     [ 2, 6, 10], 
     [ 3, 7, 11], 
     [ 4, 8, 12]]) 

Но тогда вы должны были бы дать axis=1 к mean.