Python: nump.pad() 函数的使用

import numpy as np
a = np.array([1,2,3,4])
print(a.shape)
b = [0.5,1.2,1.8,1.9,2.1,2.5,2.3,3.4,3.9,4.1]
print(b)
tmp = a[:,None] - np.array(b)
print(tmp)
# pred = np.sum(tmp < 0,axis=0)
# print(pred)
print(" ")
prob = np.pad(tmp, pad_width=((0,0),(1,1)),mode="constant",constant_values=(0,1))
print(prob)

直接看效果：