将矩阵补齐0

matrix = [[1],
          [1, 2],
          [1, 2, 3],
          [1, 2, 3, 4],
          [1, 2, 3, 4, 5],
          [3, 4, 5],
          [2, 3, 4, 5],
          ]
# 现在需要将矩阵中所有的列表长度对齐到最长的列表的长度5，末尾全部用0填充
max_len = max((len(l) for l in matrix))
new_matrix = list(map(lambda l: l + [0] * (max_len - len(l)), matrix))
print(new_matrix)