python - 在 Numpy/Scipy 中切片数组

我有一个像这样的数组:

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

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

即

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

最佳答案

二维 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]])

关于python - 在 Numpy/Scipy 中切片数组，我们在Stack Overflow上找到一个类似的问题： https://stackoverflow.com/questions/2725750/