python - 如何向 NumPy 中的矩阵添加保护环？

Question

使用 NumPy ，矩阵A有n行m列，我想给矩阵A加一个保护环，那个保护环全为0。

我该怎么办？使用 reshape ？但是这个元素不足以组成一个n+1 m+1的矩阵。

还是等等？

提前致谢

我的意思是一个额外的单元格环，它总是包含 0 个环绕矩阵 A。基本上有一个矩阵 B 有 n+2 行 m+2 列，其中第一行和第一列以及最后一行和最后一列都是零，其余的与矩阵A相同。

Answer 1

跟进您的 comment :

>>> import numpy
>>> a = numpy.array(range(9)).reshape((3,3))
>>> b = numpy.zeros(tuple(s+2 for s in a.shape), a.dtype)
>>> b[tuple(slice(1,-1) for s in a.shape)] = a
>>> b
array([[0, 0, 0, 0, 0],
       [0, 0, 1, 2, 0],
       [0, 3, 4, 5, 0],
       [0, 6, 7, 8, 0],
       [0, 0, 0, 0, 0]])