不同维度的数组，取最大行列，并分别填充每个数组，保证所有数组维度一样？

Question

不同维度的数组，取最大行列，并分别填充每个数组，保证所有数组维度一样？

Answer 1

python脚本名称为test.py:

#coding:gbk
def pad(a,pading=0):
    max_dim = max([len(_) for _ in a])
    for w in a:
        w_dim = len(w)
        pad_list = [pading] * (max_dim - w_dim)
        w.extend(pad_list)
    return a

if __name__ == '__main__':
    a = [[1,2,3],
        [4,5],
        [6,7,8,9,10]
    ]
    print("填充前：",a)
    pad(a,0)
    print("填充后: ",a)

运行命令：python test.py 后结果如下：

填充前： [[1, 2, 3], [4, 5], [6, 7, 8, 9, 10]]
填充后: [[1, 2, 3, 0, 0], [4, 5, 0, 0, 0], [6, 7, 8, 9, 10]]