2024-03-29 14:30:00 发布
网友
我有一个现有的矩阵:
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]]
非常感谢您的帮助
您可以将列表理解与列表切片结合使用:
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])
您可以将列表理解与列表切片结合使用:
输出:
简单地说,您可以迭代矩阵并对每个列表进行切片
输出:
相关问题 更多 >
编程相关推荐