python - 在 Python 中提取部分 2D-List/Matrix/List of lists

Question

我想在 Python 中提取二维列表(=列表列表)的一部分。我经常用Mathematica，写起来很方便

matrix[[2;;4,10;;13]] 

这将提取矩阵中第 2 行和第 4 行以及第 10 列和第 13 列之间的部分。

在Python中，我只是用

[x[firstcolumn:lastcolumn+1] for x in matrix[firstrow:lastrow+1]]

是否还有更优雅或更有效的方法来做到这一点？

Answer 1

你要的是numpy arrays和切片运算符 :。

>>> import numpy

>>> a = numpy.array([[1,2,3],[2,2,2],[5,5,5]])
>>> a
array([[1, 2, 3],
       [2, 2, 2],
       [5, 5, 5]])

>>> a[0:2,0:2]
array([[1, 2],
       [2, 2]])