如何用一些二维numpy数组填充二维numpy数组

Question

我有一个二维numpy数组，其大小从一个值到另一个值都不同。

如果输入值的维数小于要求的尺寸，如何填充零来概括大小？

例如：

具有numpy数组维的输入列表：
[（40，173），
（40，14），
（40，56），
（40，173）]

我希望所有数组都为（40，173），如果数组大小较小，则比用零填充其余部分。

Answer 1

这称为零填充。假设您要填充数组A：

np.pad(A, ((0, rows), (0, cols)), "constant")

您可以设置：

• rows = 40 - A.shape[0]
• cols = 173 - A.shape[1]

更多信息：Zero pad numpy array

Answer 2

分配给zeros数组：

In [160]: res = np.zeros((40,173),int)                                                               
In [161]: res[0:20, 0:30] = np.ones((20,30),int)