在 Numpy / Scipy 中切片数组

Question

我有一个像这样的数组：

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

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

IE

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

原文由 user248237 发布，翻译遵循 CC BY-SA 4.0 许可协议

Answer 1

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

>>> 从 numpy 导入数组
>>> a = array([[1,2,3],[3,4,5],[4,5,6]])
>>> 一个[:,1:]
数组([[2, 3],
       [4, 5],
       [5, 6]])

原文由 Michael Aaron Safyan 发布，翻译遵循 CC BY-SA 3.0 许可协议

Answer 2

这是你要找的吗？

 a[:,1:]

原文由 Wolph 发布，翻译遵循 CC BY-SA 2.5 许可协议