3d matrice Numpy a 2d

Question

Ho una matrice 3d simili

arange(16).reshape((4,2,2)) 
array([[[ 0, 1], 
     [ 2, 3]], 

     [[ 4, 5], 
     [ 6, 7]], 

     [[ 8, 9], 
     [10, 11]], 

     [[12, 13], 
     [14, 15]]]) 

e vorrebbe impilarle in formato griglia, finendo con

array([[ 0, 1, 4, 5], 
     [ 2, 3, 6, 7], 
     [ 8, 9, 12, 13], 
     [10, 11, 14, 15]]) 

C'è un modo di fare a meno esplicitamente hstacking (e/o vstacking) o aggiungendo una dimensione extra e rimodellamento (non è sicuro che ciò funzionerebbe)?

Grazie,

Answer 1

In [27]: x = np.arange(16).reshape((4,2,2)) 

In [28]: x.reshape(2,2,2,2).swapaxes(1,2).reshape(4,-1) 
Out[28]: 
array([[ 0, 1, 4, 5], 
     [ 2, 3, 6, 7], 
     [ 8, 9, 12, 13], 
     [10, 11, 14, 15]]) 

---

ho postato funzioni più generali per reshaping/unshaping arrays into blocks, here.