Создание новой матрицы из существующей с сохранением только первых двух столбцов в Python

Question

У меня есть существующая матрица:

    matrix = [[3,5,7,9],
              [6,8,12,14],
              [2,1,0,5],
              [7,8,9,10]]

Я пытаюсь извлечь только первые два столбца из любой матрицы и сохранить их в другой матрице.

Например, я надеюсь, что новая матрица будет выглядеть так:

   new_matrix = [[3,5],
                 [6,8],
                 [2,1],
                 [7,8]]

Любая помощь очень ценится.

Answer 1

Проще говоря, вы можете перебирать матрицу и разрезать каждый список.

matrix = [[3, 5, 7, 9],
          [6, 8, 12, 14],
          [2, 1, 0, 5],
          [7, 8, 9, 10]]
new_matrix=[]
for mat in matrix:
    new_matrix.append(mat[:2])

Выход:

[[3, 5], [6, 8], [2, 1], [7, 8]]

Answer 2

Вы можете использовать понимание списка с нарезкой списка:

matrix = [[3,5,7,9],
          [6,8,12,14],
          [2,1,0,5],
          [7,8,9,10]]
new_matrix = [i[:2] for i in matrix]

Выход:

[[3, 5], [6, 8], [2, 1], [7, 8]]