Python中如何将matrix2转换为与matrix1一致的嵌套矩阵格式

问题定位 #

你尝试将二维矩阵matrix2转换为和matrix1一致的三维结构时，存在两个核心错误：

• 误用未定义的i/j作为索引，循环变量应为x和y
• 直接取matrix2[i]会获取整行数据，而非单个元素，导致结构不匹配

正确实现代码 #

首先确保matrix2的维度与matrix1的前两维一致（即len(string0)+1行、len(string1)+1列），然后通过嵌套循环生成目标结构：

string0 = "hello"
string1 = "Good"
# 调整matrix2维度匹配string0和string1的长度+1
matrix2 = [[0, 1, 2, 3, 4], 
           [1, 2, 3, 4, 5], 
           [2, 3, 4, 5, 6], 
           [3, 4, 5, 6, 7], 
           [4, 5, 6, 7, 8], 
           [5, 6, 7, 8, 9]]

# 生成与matrix1格式完全一致的三维矩阵
tmp = [[[matrix2[x][y], matrix2[x][y]] for y in range(len(string1)+1)] for x in range(len(string0)+1)]
print(tmp)

灵活调整逻辑 #

如果需要内层列表的两个值来自matrix2的不同位置，只需修改内层的取值逻辑即可，例如：

# 示例：内层列表为matrix2的(x,y)和(y,x)位置元素
tmp = [[[matrix2[x][y], matrix2[y][x]] for y in range(len(string1)+1)] for x in range(len(string0)+1)]

注意事项 #

• matrix1的结构是(len(string0)+1) × (len(string1)+1) × 2，必须保证matrix2的行、列数与之对应，否则会触发索引越界错误
• 循环变量x对应外层行，y对应中层列，通过matrix2[x][y]精准定位二维矩阵中的单个元素

内容的提问来源于stack exchange，提问作者TheRi