使用Python列表创建单位矩阵时出现错误,期望结果不符求排查
问题分析与修复方案
嘿,我来帮你排查这个问题!你得到全1矩阵的核心原因是代码里重复添加了同一个列表的引用到identity中,导致所有子列表其实都是同一个对象。
问题出在哪?
看你这段代码:
null=[] for i in range(3): null.append(i*0) for j in range(3): identity.append(null)
你先创建了一个null = [0,0,0]列表,然后连续三次把它添加到identity里。这时候identity里的三个元素并不是三个独立的列表,而是指向同一个null列表的引用。所以当你执行identity[k][k]=1时,其实是在修改同一个底层列表——三次修改后,这个列表就变成了[1,1,1],最后identity里的三个元素自然都是它。
修复后的代码
这里给你两种可行的修复方式:
方式一:每次循环创建新的子列表
identity = [] for j in range(3): # 每次循环生成一个新的全0列表,避免复用引用 null = [0 for _ in range(3)] identity.append(null) for k in range(3): identity[k][k] = 1 print(identity)
方式二:用列表推导式更简洁高效
直接通过嵌套列表推导式一步生成单位矩阵,逻辑更清晰:
identity = [[1 if i == j else 0 for j in range(3)] for i in range(3)] print(identity)
运行这两种代码都会输出你期望的标准单位矩阵:[[1, 0, 0], [0, 1, 0], [0, 0, 1]]
内容的提问来源于stack exchange,提问作者Charles
相关产品推荐
相关产品推荐

