手动实现矩阵转置:向空列表的列表添加元素仅保留最后值的问题
解决手动实现矩阵转置的问题
哦,我一眼就发现问题所在啦——你把bacon0的初始化代码放到嵌套循环内部了!每次循环迭代时,都会重新创建一个全新的空列表bacon0,之前循环中添加的所有元素都会被直接覆盖,到最后自然就只剩下最后一次循环添加的['c']了。
修正后的代码
我们只需要把bacon0的初始化移到循环外面,先根据原矩阵的列数创建好对应的空列表结构,再在循环里逐个填充元素:
spam=[[1,2,3,'a'], [4,5,6,'b'], [7,8,9,'c']] # 先初始化bacon0:原矩阵有多少列,就创建多少个空列表 bacon0 = [[] for n in range(len(spam[0]))] for j in range(len(spam)): for i in range(len(spam[0])): # 直接往对应列的子列表里追加当前元素 bacon0[i].append(spam[j][i]) print(bacon0) # 输出:[[1,4,7],[2,5,8],[3,6,9],['a','b','c']]
更直观的写法(可选)
如果觉得嵌套索引有点绕,我们可以换个逻辑:先遍历每一列的索引,再遍历每一行提取该列的元素,最后把整列元素加入结果列表,这样逻辑更清晰:
spam=[[1,2,3,'a'], [4,5,6,'b'], [7,8,9,'c']] bacon0 = [] # 遍历每一列的索引 for col_idx in range(len(spam[0])): current_col = [] # 遍历每一行,提取当前列的元素 for row in spam: current_col.append(row[col_idx]) bacon0.append(current_col) print(bacon0)
这样就能得到你预期的转置结果啦!
内容的提问来源于stack exchange,提问作者r0gu3cic
相关产品推荐
相关产品推荐

