如何使用numpy.pad

时间:2019-11-30 12:39:52

标签: python arrays numpy

我有一个形状为(18, 300)的数组。最后,无论数组shape[0]是什么,我都希望通过用零数组填充间隙来获得形状为(20, 300)的数据(现在忽略shape[0]为大于20)。有几个问题,但我做不到。到目前为止,我已经尝试过:

a = np.random.rand(18, 300)

gap = 20 - 18 # this is hardoced but it will be generalized in real file scenario

np.pad(a, (0, gap), 'constant', constant_values=(0, np.zeros(300)))

我认为这应该将2D数组从(18,300)扩展为(20,300),并用300的零填充。但是我得到一个错误:

  

无法将形状(300)的输入数组广播到形状(20,2)

此错误是什么意思?如何用300的零填充18到20?

谢谢。

0 个答案:

没有答案