python - 将零值添加到一维数组并将其拟合到二维数组中，这样就不会引发错误

Question

我在 for 循环中有可变大小的一维数组，我将这些数组放在大小为 40x4000 的二维数组的行中。如何将零附加到一维数组的末尾，以使最终大小变为 1x4000 并且能够适合二维数组？

例如。

size of 1-D: 3700 , so add 300 zeros at the end
size of 1-D: 3800 , so add 200 zeros at the end

for i in range(n)
   s = func(i) #returns row vector
   # what to do here?
   two_dim[i] = s

编辑：这些是 numpy 数组而不是列表

Answer 1

为较小的输入执行此操作，希望这就是您正在寻找的。

对于初始列表testList = [9, 9, 9, 8, 4, 5]

>>> maxListLen = 10
>>> testList = testList+[0]*(maxListLen-len(testList))
[9, 9, 9, 8, 4, 5, 0, 0, 0, 0]

由于 OP 正在寻找numpy数组。

>>> testArray = numpy.array([1, 2, 3, 4])
>>> testArray = numpy.append(testArray, [0]*(maxListLen-len(testArray)))
>>> testArray
array([1, 2, 3, 4, 0, 0, 0, 0, 0, 0])