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

2024-03-29 14:30:00 发布

您现在位置: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]]

非常感谢您的帮助


Tags: new矩阵matrix
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]]

简单地说,您可以迭代矩阵并对每个列表进行切片

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

相关问题 更多 >