将新列添加到列表的元素

Question

我有一个包含其他数字列表的列表（input_m）。我想为列表的每个元素添加一个新列（零）以获得最终列表（output_m）：

input_m = [[3, 2], [5, 1], [4, 7]]
output_m = [[3, 2, 0], [5, 1, 0], [4, 7, 0]]

我尝试过列表推导，但是我没有得到我想要的格式的output_m。这是代码：

def add_column(matrix):
    res = [[item,0] for item in matrix]
    return res

output_m = add_column(input_m)

output_m = [[[3, 2], 0], [[5, 1], 0], [[4, 7], 0]]

有任何帮助吗？感谢

Answer 1

在列表理解中 -

res = [[item,0] for item in matrix]

item是列表，这就是您获得结果的原因 - [[[3, 2], 0], [[5, 1], 0], [[4, 7], 0]]

尝试连接，例如 -

def add_column(matrix):
    res = [item + [0] for item in matrix]
    return res