[已解决]如何获取二维列表斜对角上的数据

九歌当下

1. >>> matrix = [[1, 2, 3],
   
2.               [4, 5, 6],
   
3.               [7, 8, 9]]
   
4. >>> diag = [matrix[i][i] for i in range(len(matrix))]
   
5. >>> diag
   
6. [1, 5, 9]

复制代码

从左到右的对角数据会获取，从右到左的怎么办

最佳答案

月排行榜 / 总排行榜

tommyyu

2022-8-15 13:08:00

1. diag = [matrix[i][3-i] for i in range(len(matrix))]

复制代码

跳转到最佳答案楼层

sj185123

1. matrix = [[1, 2, 3],
   
2.           [4, 5, 6],
   
3.           [7, 8, 9]]
   
4. l = len(matrix)
   
5. diag = [matrix[i][l-i-1] for i in range(l)]
   
6. print(diag)

复制代码

tommyyu

1. diag = [matrix[i][3-i] for i in range(len(matrix))]

复制代码

九歌当下

tommyyu 发表于 2022-8-15 13:08

谢谢了

九歌当下

tommyyu 发表于 2022-8-15 13:08

1. >>> diag = [matrix[i][3-i] for i in range(len(matrix))]
   
2. Traceback (most recent call last):
   
3.   File "<pyshell#1>", line 1, in <module>
   
4.     diag = [matrix[i][3-i] for i in range(len(matrix))]
   
5.   File "<pyshell#1>", line 1, in <listcomp>
   
6.     diag = [matrix[i][3-i] for i in range(len(matrix))]
   
7. IndexError: list index out of range

复制代码

sj185123

tommyyu 发表于 2022-8-15 13:08

错了吧，3-i-1吧

tommyyu

九歌当下 发表于 2022-8-15 13:21

写错了，是3-i-1

九歌当下

1. diag = [matrix[i][list(reversed(range(len(matrix))))[i]] for i in range(len(matrix))]

复制代码

九歌当下

tommyyu 发表于 2022-8-15 13:39
写错了，是3-i-1

好的，谢谢了