如何添加新的行和列以填充numpy数组

时间:2019-03-18 17:41:50

标签: python numpy vectorization

我有一组矩阵,所有矩阵都有不同的大小/尺寸,我需要将它们全部填充为相同的大小(805,16866)。贝娄是我目前正在使用的代码,但是我相信它可以被矢量化,但是我不确定如何做到这一点。

x1 = np.zeros((805, 16866))

for i in range(x[0].shape[0]):
    for j in range(x[0].shape[1]):
        x1[i, j] = x[0][i, j]

2 个答案:

答案 0 :(得分:2)

在上面的示例中,x1[:x[0].shape[0], :x[0].shape[1]] = x应该有效

答案 1 :(得分:0)

numpy.pad函数(docs)可以满足您的需求。

您想要的代码是

x1 = np.zeros((805,16866))

for i in range(x.shape[0])
  x1[i,:]= np.pad(x[i,:],x1.shape[0],'constant',constant_values=0)