在现有数组周围添加边框（用 0 填充）

Question

我正在尝试在现有数组周围添加一个边框（用 0 填充）。 我的代码：

a = np.random.random((4, 5))
m, n = a.shape
b = np.zeros((m+1, n+1))
b[1:-1,1:-1] = a
a,b

但我得到了一个错误： ValueError: could not broadcast input array from shape (4,5) into shape (3,4)

为什么我不能直接为 2D numpy 数组的某些部分赋值？问题出在哪里？

Answer 1

您可以使用 np.pad 功能完成您的任务：

b = np.pad(a, 1)

比你的代码更短。

阅读此函数的文档以了解任何详细信息。