numpy多维数组获取指定矩形区域

       ndarray中,当我们想获取其指定行列的子矩形局域时,该如何获取?一般的,我们容易误认为通过传入相应索引的数组来直接获取,但是结果出乎我们预期,这时我们得到的却是相应数组构成的坐标对应于ndarray中的元素构成的一维数组。如下所示。

上面例子中,我们想获取arr对象中的右下角四个元素构成的矩形子区域,所以我们尝试通过arr[[1,2],[1,2]]的方式获取,结果返回的是(1,1),(2,2)坐标对应的元素,并不是我们想要的矩形区域。实际上,对于numpy的ndarray,当我们对行和列都用数组索引时(这种方式成为fancy索引),其返回的并不是相应的行列构成的子区域,而是传入数组构成的坐标对应的元素。如果我们想要获取相应的矩形区域,则可以通过下列方式。即先获取相应的行,再通过切片索引和数组索引获取。

       实际上,在ndarray中,只要行和列其中有一方不是数组索引,或者说只要有一方是切片索引,那么获取到的就是相应的子矩形区域,如下所示。

       最后要说明一下的是,虽然pandas是基于Numpy的,但是对于DataFrame的索引,即使行列都是数组索引,返回的是相应矩形区域的DataFrame,这点和ndarray并不一样。

猜你喜欢

转载自blog.csdn.net/S_o_l_o_n/article/details/108610596
今日推荐