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)

我怎么能这样做，因为数据结构不是 flat ？你建议先把它展平吗？

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

Answer 1

您可以使用负止损指数来排除最后x行/列和正常起始指数：

>>> 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：提取2D numpy数组的核心

1 个答案: