在Python中提取2D列表/矩陣/列表的一部分

Question

我想在Python中提取二維列表（=列表列表）的一部分。我用數學很多，而且有非常方便寫

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