Wie behalten wir die spezifische Dimension unverändert bei der Umformung in numpiger Form?

Question

Ich habe eine riesige (N * 20) Matrix, wo alle 5 Zeilen ein gültiges Beispiel ist, dh. jede (5 * 20) Matrix. Ich versuche, es in eine (N/5,1,20,5) Matrix umzuformen, in der die Dimension 20 unverändert bleibt. Ich könnte es in tensroflow mit keep_dim tun, aber wie kann ich das in numpy erreichen?

Vielen Dank im Voraus.

Answer 1

umformen und dann die Achsen tauschen um:

arr1 = arr.reshape(N/5,5,1,20) 
arr2 = arr1.transpose(0,2,3,1) 

zum Beispiel

In [476]: arr = np.arange(24).reshape(6,4) 
In [477]: arr 
Out[477]: 
array([[ 0, 1, 2, 3], 
     [ 4, 5, 6, 7], 
     [ 8, 9, 10, 11], 
     [12, 13, 14, 15], 
     [16, 17, 18, 19], 
     [20, 21, 22, 23]]) 
In [478]: arr1 = arr.reshape(2,3,1,4) 
In [479]: arr2 = arr1.transpose(0,2,3,1) 
In [480]: arr2.shape 
Out[480]: (2, 1, 4, 3) 
In [482]: arr2 
Out[482]: 
array([[[[ 0, 4, 8], 
     [ 1, 5, 9], 
     [ 2, 6, 10], 
     [ 3, 7, 11]]], 


     [[[12, 16, 20], 
     [13, 17, 21], 
     [14, 18, 22], 
     [15, 19, 23]]]])