python - 如何从另一个二维数组中分割一个二维数组并获得其余部分

Question

我有一个 numpy 数组A = np.array([[0, 1, 0, 0, 0],[0, 0, 1, 0, 0],[1, 0, 1, 1, 1]])

我想B = np.array([[0, 1, 0],[0, 0, 1],[1, 0, 1]])从 A 切片并将 A 的其余部分作为一个 numpy 数组获取C = np.array([[0, 0], [0, 0],[1,1]])

我正在寻找一个通用代码，因为我的矩阵 B 在 A 内部的不同迭代中发生变化。

谢谢

Answer 1

在 numpy 中，这实际上很容易，就像你可以为这个问题做的那样：

import numpy as np
k = 3 #k is used to slice the array, here k = 3
A = np.array([[0, 1, 0, 0, 0],[0, 0, 1, 0, 0],[1, 0, 1, 1, 1]])
C = np.array([a[-i:] for i in range(k, len(a)])
A = np.array([a[i:i+k] for i in range(0, len(a), k)])

希望这可以帮助。