Python矩阵转置：zip(*matrix)与列表推导式对比

Python矩阵转置看似简单，实则暗藏陷阱：`zip(*matrix)`虽简洁高效却默认返回不可变元组，需手动转为列表才能修改；而直观的嵌套列表推导式在空矩阵或不规则结构下直接崩溃，且性能更差；真正稳健的做法是先校验输入（是否为空、行列是否一致），再用`[list(r) for r in zip(*matrix)]`安全转置；若已使用NumPy，则应直接调用`.T`或`.transpose()`——它零拷贝、保类型、支持高维，远胜原生方案。选择哪种方法，关键不在“怎么写”，而在于你是否清楚当前数据的可信度、规模和后续用途。

用 zip(*matrix) 转置时，为什么结果是元组？

因为 zip 默认把每行对应位置的元素打包成 tuple，不是 list。你拿到的是 [('a', 'd'), ('b', 'e'), ('c', 'f')] 这种结构，不是想要的嵌套列表。

• 如果后续要修改元素（比如 result[0][0] = 'x'），会报 TypeError: 'tuple' object does not support item assignment
• 常见修复方式是套一层 list： [list(row) for row in zip(*matrix)]
• 注意：若原矩阵为空（matrix = []），zip(*matrix) 返回空迭代器，直接转 list 得到 []，逻辑上正确但容易被忽略边界

列表推导式 [[row[i] for row in matrix] for i in range(len(matrix[0]))] 的坑

这写法看似直观，但隐含两个硬性前提：矩阵非空、每行长度一致。一旦不满足，运行时就崩。

• 若 matrix = []，matrix[0] 直接抛 IndexError: list index out of range
• 若 matrix = [[1, 2], [3]]（不规则），row[i] 在第二轮访问 [3][1] 时报错
• 性能上，它遍历矩阵两次（外层 range + 内层 for row），而 zip 是单次迭代，大数据量时差异明显

真正安全又通用的写法：先校验再转置

生产代码里别赌输入格式。哪怕只是脚本，也值得加两行防御。

• 检查空矩阵：if not matrix: return []
• 检查行列一致性（可选但推荐）：if not all(len(row) == len(matrix[0]) for row in matrix): raise ValueError("Matrix rows have inconsistent lengths")
• 然后用 [list(r) for r in zip(*matrix)] —— 简洁、可读、无副作用

NumPy 用户别硬套 Python 原生方案

如果你已经在用 numpy，matrix.T 或 matrix.transpose() 是零拷贝、支持任意维度、自动处理 dtype 的正解。

• 原生 zip 或推导式对大数组慢几个数量级，且无法保留数值类型（全变成 Python int/float）
• 混用时注意：传给 np.array() 前，确保嵌套结构规整，否则会降维成一维 object 数组
• 错误示范：np.array([[1,2], [3]]) → shape (2,)，不是 (2, 2)

以上就是本文的全部内容了，是否有顺利帮助你解决问题？若是能给你带来学习上的帮助，请大家多多支持golang学习网！更多关于文章的相关知识，也可关注golang学习网公众号。