您可以使用
np.pad它,它也可以
2-D使用指定填充宽度的元组填充数组
((top,bottom), (left, right))。为此,您可以定义:
def pad_to_length(x, m): return np.pad(x,((0, 0), (0, m - x.shape[1])), mode = 'constant')
用法
您可以从查找
ndarray具有最高列数的开始。假设您有两个,
a并且
b:
a = np.array([[0, 1, 2], [3, 4, 5], [6, 7, 8]])b = np.array([[ 0, 1, 2, 3, 4], [ 5, 6, 7, 8, 9], [10, 11, 12, 13, 14]])m = max(i.shape[1] for i in [a,b])# 5
然后使用此参数填充
ndarrays:
pad_to_length(a, m)array([[0, 1, 2, 0, 0], [3, 4, 5, 0, 0], [6, 7, 8, 0, 0]])



