零填充numpy数组

零填充numpy数组

numpy.pad

使用

constant

mode可以满足您的需要，在这里我们可以传递一个元组作为第二个参数来告诉每个大小要填充多少个零

(2,3)

，例如a将在左边填充 2个 零，在右边填充 3个 零：

给出

A

为：

A = np.array([1,2,3,4,5])np.pad(A, (2, 3), 'constant')# array([0, 0, 1, 2, 3, 4, 5, 0, 0, 0])

也可以通过将元组的元组作为填充宽度来填充2D numpy数组，其格式为

((top, bottom), (left, right))

：

A = np.array([[1,2],[3,4]])np.pad(A, ((1,2),(2,1)), 'constant')#array([[0, 0, 0, 0, 0],# 1 zero padded to the top#       [0, 0, 1, 2, 0],# 2 zeros padded to the bottom#       [0, 0, 3, 4, 0],# 2 zeros padded to the left#       [0, 0, 0, 0, 0],# 1 zero padded to the right#       [0, 0, 0, 0, 0]])

---

对于您的情况，您可以将左侧垫板指定为零，而将右侧垫板指定为根据模块除法计算：

B = np.pad(A, (0, 1024 - len(A)%1024), 'constant')B# array([1, 2, 3, ..., 0, 0, 0])len(B)# 1024

对于较大的

A

：

A = np.ones(3000)B = np.pad(A, (0, 1024 - len(A)%1024), 'constant')B# array([ 1.,  1.,  1., ...,  0.,  0.,  0.])len(B)# 3072