Làm thế nào để trích xuất một mảng phụ từ một mảng 2d không rõ ràng?

Question

Tôi muốn trích xuất một mảng có khối lượng với kích thước được chỉ định từ mảng 2d có khối lượng - về cơ bản tôi muốn cắt mảng. Ví dụ, nếu có một mảng NumPy như thế này:

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

Tôi muốn trích xuất một 2x2 từ nó và kết quả nên là:

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

Làm thế nào tôi có thể làm điều đó?

Answer 1

Với mảng này:

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

Bạn có thể cắt nó dọc theo hai khía cạnh:

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