Python：提取二维numpy数组的核心

Question

假设我有一个像这样的 2D numpy 数组：

In[1]: x
Out[1]:
array([[0, 0, 0, 0, 0],
       [1, 1, 1, 1, 1],
       [2, 2, 2, 2, 2],
       [3, 3, 3, 3, 3],
       [4, 4, 4, 4, 4],
       [5, 5, 5, 5, 5]], dtype=int64)

我想提取 (n-1)*(m-1) 核心，即：

array([[1, 1, 1],
       [2, 2, 2],
       [3, 3, 3],
       [4, 4, 4]], dtype=int64)

我该怎么做，因为数据结构不是 平坦的？你建议先把它弄平吗？

这是一个更大数组的简化版本，其核心维度为 (n-33)*(n-33)。

Answer 1

您可以使用负停止索引来排除最后一个 x rows/columns 和正常开始索引：

>>> x[1:-1, 1:-1]
array([[1, 1, 1],
       [2, 2, 2],
       [3, 3, 3]], dtype=int64)

对于您的新示例：

>>> t = np.array([[0, 0, 0, 0, 0],
                  [1, 1, 1, 1, 1],
                  [2, 2, 2, 2, 2],
                  [3, 3, 3, 3, 3],
                  [4, 4, 4, 4, 4],
                  [5, 5, 5, 5, 5]], dtype=np.int64)
>>> t[1:-1, 1:-1]
array([[1, 1, 1],
       [2, 2, 2],
       [3, 3, 3],
       [4, 4, 4]], dtype=int64)

您还可以删除 2 个前导列和尾随列：

>>> t[1:-1, 2:-2]
array([[1],
       [2],
       [3],
       [4]], dtype=int64)

或行：

>>> t[2:-2, 1:-1]
array([[2, 2, 2],
       [3, 3, 3]], dtype=int64)

Python：提取二维numpy数组的核心

Python: extract the core of a 2D numpy array

python

arrays

numpy

slice