在numpy矩阵中切割特定的行和列

Question

我有一个列表，我将其转换为矩阵。

m = [[0, 2, 1, 0, 0], [0, 0, 0, 3, 4], [0, 0, 0, 0, 0], [0, 0, 0, 0, 0], [0, 0, 0, 0, 0]] 

n = np.matrix(m) 

我该如何去创建一个基于前三行和列的新矩阵？

具体做法是：

I = [[0, 2, 1], [0, 0, 0], [0, 0, 0]] 

我想下面的行会工作

I = m[0:3, 0:3] 

，但我得到

TypeError: list indices must be integers or slices, not tuple 

Answer 1

你是不是切片的矩阵的一个列表中的错误。列表不能将元组作为参数。 使用n[0:3,0:3]而不是m[0:3, 0:3]。

Answer 2

除了user1753919的回答，您可以用下面的代码片名单：

In [10]: [row[0:3] for row in m[0:3]] 
Out[10]: [[0, 2, 1], [0, 0, 0], [0, 0, 0]] 

不过我建议你这样做与矩阵。因为它更简单。