时间戳

首页 > 解决方案 > 从现有矩阵创建新矩阵，同时仅将前两列存储在 python 中

python - 从现有矩阵创建新矩阵，同时仅将前两列存储在 python 中

问题描述

我有一个现有的矩阵：

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

任何帮助深表感谢。

标签： pythonlistmatrix

解决方案

您可以将列表推导与列表切片一起使用：

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

推荐阅读