如何用Python创建元素连续递增的2×3二维矩阵?
如何生成元素连续递增的2×3 Python二维矩阵?
问题说明
需要创建一个2×3的二维矩阵,元素从1开始依次递增,期望输出为[[1,2,3],[4,5,6]],但现有代码运行后输出[[2, 3, 4], [2, 3, 4]],不符合预期。
错误原因分析
原代码存在两个核心问题:
- 外层循环中每次都将
i重置为1,导致每一行的计数起点完全相同,无法衔接上一行的末尾数值 - 内层循环从
i+1开始添加元素,直接跳过了起始值1,导致每行的第一个元素都是2
另外,代码开头的rows = int(input())和cols = int(input())被后续的赋值覆盖,属于冗余代码,若不需要用户输入可直接删除。
修正方案
这里提供两种简洁有效的实现方式:
方法1:使用全局计数器(直观易懂)
通过维护一个全局计数器,确保元素从1开始连续递增:
rows = 2 cols = 3 matrix = [] current_num = 1 # 初始化起始数值为1 for _ in range(rows): row = [] for _ in range(cols): row.append(current_num) current_num += 1 # 每添加一个元素,计数器自增 matrix.append(row) print(matrix) # 输出: [[1, 2, 3], [4, 5, 6]]
方法2:通过数学公式计算(无额外变量)
利用行号和列号的关系直接计算元素值,无需额外计数器:
rows = 2 cols = 3 matrix = [] for i in range(rows): row = [] for j in range(cols): # 第i行第j列的元素 = 行号×列数 + 列号 + 1 row.append(i * cols + j + 1) matrix.append(row) print(matrix) # 输出: [[1, 2, 3], [4, 5, 6]]
内容的提问来源于stack exchange,提问作者sim
相关产品推荐
相关产品推荐

