在 Numpy / Scipy 中切片数组

Question

我有一个数组，如：

a = array([[1,2,3],[3,4,5],[4,5,6]])

从中切出只有前两列“a”的 1x2 数组的最有效方法是什么？

IE

array([[2,3],[4,5],[5,6]]) in this case.

Answer 1

二维 numpy 数组使用a[i,j] （不是a[i][j] ）进行索引，但是您可以对 numpy 数组和矩阵使用相同的切片符号，就像在 python 中使用普通矩阵一样（只需将它们放入单个[] )：

>>> from numpy import array
>>> a = array([[1,2,3],[3,4,5],[4,5,6]])
>>> a[:,1:]
array([[2, 3],
       [4, 5],
       [5, 6]])

Answer 2

这是你要找的吗？

a[:,1:]

Answer 3

引用文档，基本的切片语法是i:j:k ，其中i是起始索引， j是停止索引， k是步长（当k > 0 ）。

现在如果i没有给出，如果k > 0 ，它默认为k > 0 。 否则，对于k < 0 ， i默认为n - 1 （其中n是数组的长度）。

如果未给出j ，则默认为n （数组长度）。

那是一维数组。

现在二维数组是一个不同的野兽。 其切片语法是a[rowrange, columnrange] 。

因此，如果您想要所有行，但只想要最后两列，就像您的情况一样，您可以：

a[0:3, 1:3]

这里，“ [0:3] ”表示从0到3的所有行，“ [1:3] ”表示从第1列到第3列的所有列。

现在您可能想知道，即使您只有 3 列并且编号从 1 开始，它也必须返回 3 列，对吗？ 即：第 1 列、第 2 列、第 3 列

这是这个语法的棘手部分。 第一列实际上是第 0 列。所以当您说“ [1:3] ”时，您实际上是在说给我第 1 列和第 2 列。这是您想要的最后两列。 （实际上没有第 3 列。）

现在如果你不知道你的矩阵有多长，或者你想要所有的行，你可以把那部分留空。 IE

a[:, 1:3]

列也一样。 即如果你想说，除了第一行之外的所有列，你会写

a[0:1,:]

现在，上面的答案a[:,1:]是如何工作的，因为当你对列说“ [1:] ”时，这意味着给我除了第 0 列之外的所有内容，直到所有列的末尾。 即空意味着“直到最后”。

现在你必须意识到逗号两边的任何东西都是我上面第一次提到的一维情况的子集。 即如果你想使用步长指定你的行，你可以写

a[::2,1]

在你的情况下哪个会返回

array([[2, 3],
       [5, 6]])

即a[::2,1]阐明为：每隔一行给我，从最上面开始，只给我第二列。

这花了我一些时间才弄明白。 所以把它贴在这里，以防万一它对某人有帮助。