二维数组 - 选择列范围

Question

我有一个二维的numpy数组 A，它的大小是MxN。我想选择一段列，比如从m到n的列（或者行也可以，没关系），像这样：

A = A[m-n,:]

有没有简单的方法可以做到这一点呢？

Answer 1

你可以这样做：

B = A[m:n,:]

或者如果你想包含第n行的话：

B = A[m:n+1,:]

列的操作也是类似的：

C = A[:,m:n]

或者

C = A[:,m:n+1]