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

九歌当下

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

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

最佳答案

月排行榜 / 总排行榜

tommyyu

2022-8-15 13:08:00

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

跳转到最佳答案楼层

sj185123

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

tommyyu

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

九歌当下

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

谢谢了

九歌当下

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

>>> diag = [matrix[i][3-i] for i in range(len(matrix))]
Traceback (most recent call last):
  File "<pyshell#1>", line 1, in <module>
    diag = [matrix[i][3-i] for i in range(len(matrix))]
  File "<pyshell#1>", line 1, in <listcomp>
    diag = [matrix[i][3-i] for i in range(len(matrix))]
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

九歌当下

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

九歌当下

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

好的，谢谢了