You need to enable JavaScript to run this app.
优惠活动
大模型
产品
解决方案
定价
更多

Python中从现有矩阵提取前两列生成新矩阵的实现方法

提取矩阵前两列的Python实现方案

针对你给出的矩阵需求,这里有几种简洁实用的实现方式,都能精准得到预期的new_matrix:

方法一:列表推导式(最常用推荐)

这是Python里处理这类矩阵列提取问题最直观高效的写法,直接遍历原矩阵的每一行,通过切片取前两个元素:

matrix = [[3,5,7,9], [6,8,12,14], [2,1,0,5], [7,8,9,10]]
new_matrix = [row[:2] for row in matrix]
print(new_matrix)  # 输出: [[3,5], [6,8], [2,1], [7,8]]

解释:row[:2]是Python的切片语法,会截取当前行的前两个元素;遍历原矩阵的每一行后,自动拼接成只保留前两列的新矩阵。

方法二:函数式风格(map+匿名函数)

如果你偏好函数式编程的写法,可以用map配合匿名函数来实现:

matrix = [[3,5,7,9], [6,8,12,14], [2,1,0,5], [7,8,9,10]]
new_matrix = list(map(lambda row: row[:2], matrix))
print(new_matrix)

这里map会把匿名函数应用到矩阵的每一行,最后通过list()转换为列表,就能得到目标结果。

方法三:循环遍历(新手友好)

如果想更清晰地拆解逻辑,适合用普通for循环逐个构建新矩阵:

matrix = [[3,5,7,9], [6,8,12,14], [2,1,0,5], [7,8,9,10]]
new_matrix = []
for row in matrix:
    # 取当前行的第一个和第二个元素,组成新行后加入新矩阵
    new_row = [row[0], row[1]]
    new_matrix.append(new_row)
print(new_matrix)

这种方式步骤明确,能直观看到每一行的处理过程,适合刚接触Python列表操作的朋友理解底层逻辑。

以上三种方法都能完美实现需求,其中列表推导式因简洁高效,是日常开发中最常用的写法。

内容的提问来源于stack exchange,提问作者jakebrah

相关产品推荐
方舟 Agent Plan

超全模态模型 × Harness 升级,最新支持 Deepseek-V4.1-Flash、GLM-5.3 系列、Doubao-Seedream-5.0-pro、Kimi-K3 (部分), 限时 9.9 元起

最近更新时间:2026.05.26 10:05:43