numpy - numpy中有一个简单的垫子吗？

Question

是否有一个以这种方式填充数组的 numpy 函数？

import numpy as np

def pad(x, length):
    tmp = np.zeros((length,))
    tmp[:x.shape[0]] = x
    return tmp

x = np.array([1,2,3])
print pad(x, 5)

输出：

[ 1.  2.  3.  0.  0.]

我找不到办法做到这一点numpy.pad()

Answer 1

您可以使用ndarray.resize()：

>>> x = np.array([1,2,3])
>>> x.resize(5)
>>> x
array([1, 2, 3, 0, 0])

请注意，此函数的行为不同于numpy.resize()，后者使用数组本身的重复副本填充。（一致性适用于无法记住所有内容的人。）

Answer 2

Sven Marnach 的使用建议ndarray.resize()可能是最简单的方法，但为了完整起见，以下是如何使用numpy.pad：

In [13]: x
Out[13]: array([1, 2, 3])

In [14]: np.pad(x, [0, 5-x.size], mode='constant')
Out[14]: array([1, 2, 3, 0, 0])