numpy數組補齊0 numpy.pad使用方法

import numpy as np
array = np.array([1, 1, 1])

# (1,2)表示在一維數組array前面填充1位，最后面填充2位
#  constant_values=(0,2) 表示前面填充0，后面填充2
ndarray=np.pad(array,(1,2),'constant', constant_values=(0,2)) 

print("array",array)
print("ndarray=",ndarray)

　　結果

d = np.array([4,3,2,1])
d = np.pad(d,(0,2),'constant',constant_values=(0,0))
print(d)