Добавить новый столбец в элементы списка

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]]

Try конкатенации вместо Пример -

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