python初始化数组为零_python – 增加数组大小并将新元素初始化为零

在Python中,如果输入是一个numpy数组,你可以使用

np.lib.pad在它周围填充零 –

import numpy as np

A = np.array([[1, 2 ],[2, 3]]) # Input

A_new = np.lib.pad(A, ((0,1),(0,2)), 'constant', constant_values=(0)) # Output

样品运行 –

In [7]: A # Input: A numpy array

Out[7]:

array([[1, 2],

[2, 3]])

In [8]: np.lib.pad(A, ((0,1),(0,2)), 'constant', constant_values=(0))

Out[8]:

array([[1, 2, 0, 0],

[2, 3, 0, 0],

[0, 0, 0, 0]]) # Zero padded numpy array

如果你不想计算要填充多少个零的数学,你可以让代码在给定输出数组大小的情况下为你做 –

In [29]: A

Out[29]:

array([[1, 2],

[2, 3]])

In [30]: new_shape = (3,4)

In [31]: shape_diff = np.array(new_shape) - np.array(A.shape)

In [32]: np.lib.pad(A, ((0,shape_diff[0]),(0,shape_diff[1])),

'constant', constant_values=(0))

Out[32]: