2016-02-28 20 views
6

Vorrei estrarre una matrice numpy con una dimensione specificata da una matrice 2d numpy: essenzialmente voglio ritagliare la matrice. Per esempio, se dispone di un array NumPy come questo:Come si estrae un sub-array da un array 2d numerico?

([1,2,3], 
[4,5,6], 
[7,8,9]) 

vorrei estrarre una 2x2 da esso e il risultato dovrebbe essere:

([1,2], 
[4,5]) 

Come posso fare questo?

+0

È possibile usare [indicizzazione Numpy] (http://docs.scipy.org/doc/numpy /reference/arrays.ndarray.html) array [: 2,: 2] – Kasramvd

risposta

9

Data questa matrice:

>>> a 
array([[1, 2, 3], 
     [4, 5, 6], 
     [7, 8, 9]]) 

È possibile tagliare lungo entrambe le dimensioni:

>>> a[:2,:2] 
array([[1, 2], 
     [4, 5]]) 
Problemi correlati