从现有矩阵创建新矩阵，同时仅在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 = [i[:2] for i in matrix]

输出：

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