我想从numpy 2d数组中提取具有指定大小的numpy数组 - 基本上我想裁剪数组。 例如,如果有一个像这样的numpy数组:
([1,2,3],
[4,5,6],
[7,8,9])
我想从中提取2x2,结果应为:
([1,2],
[4,5])
我该怎么做?
答案 0 :(得分:22)
鉴于此数组:
>>> a
array([[1, 2, 3],
[4, 5, 6],
[7, 8, 9]])
您可以沿两个维度对其进行切片:
>>> a[:2,:2]
array([[1, 2],
[4, 5]])