用numpy填充矩阵

时间:2018-04-10 13:18:32

标签: python-3.x numpy

我有一个名为adj=dim(16,16)的2D numpy数组。我想用零填充new_adj=dim(31,31)

我试过......

new_adj=np.pad(adj,((15,31),(31,15)),mode='constant')

然而

new_adj.shape=(62, 62)

我应该得到......

new_adj.shape=(31, 31)

1 个答案:

答案 0 :(得分:3)

如果查看np.pad的文档,它会解释第二个参数中的每个元组指定在每个维度的开头和结尾添加多少个pad的位置。您在顶部添加15行,在底部添加31行,在左侧添加31列,在右侧添加15列,因此最终(62,62)矩阵。如果您只想在底部和右侧添加行和列,请执行以下操作:

new_adj = np.pad(adj, [(0, 15), (0, 15)], mode='constant')