我遇到了使用 itertools 从矩阵(列表列表)获取所需元素的问题。 我可以使用简单的 for 循环轻松完成此操作,但我对更有效的方式感兴趣。
所以我有一个列表列表(让我们称之为矩阵)和索引列表,这些索引引用矩阵中每个列表中所需的元素 像这样的东西:
matrix = [[0,1,2],[3,4,5],[6,7,8],[9,10,11],[12,13,14]]
line = [0,1,2,0,0]
所以预期的输出应该是可迭代的:0 4 8 9 12
谢谢
最佳答案
您也可以使用枚举
matrix = [[0,1,2],[3,4,5],[6,7,8],[9,10,11],[12,13,14]]
line = [0,1,2,0,0]
res = [matrix[count][i] for count, i in enumerate(line) ]
关于python - 在Python中使用itertools帮助有效地获取矩阵的元素,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/35200117/