如何设置不在矩阵子矩阵中的数

Question

假设我有一个numpy数组。我想为不在子矩阵掩码中的索引设置0。你知道吗

import numpy as np

mtrx = np.arange(25).reshape(5,5)
mask = mtrx[1:4, 1:4]

当然这是错误的：

mtrx[~mask] = 0

我想要的结果是：

[[ 0  0  0  0  0]
 [ 0  6  7  8  0]
 [ 0 11 12 13  0]
 [ 0 16 17 18  0]
 [ 0  0  0  0  0]]

Answer 1

一种方法是索引一个由零组成的数组，并从索引的原始数组中添加值：

s = np.s_[1:4,1:4]
out = np.zeros_like(mtrx)
out[s] = mtrx[s]

---

print(out)

array([[ 0.,  0.,  0.,  0.,  0.],
       [ 0.,  6.,  7.,  8.,  0.],
       [ 0., 11., 12., 13.,  0.],
       [ 0., 16., 17., 18.,  0.],
       [ 0.,  0.,  0.,  0.,  0.]])