如何从python中的列表列表中提取所有列和对角线?

2024-03-28 09:44:24 发布

您现在位置:Python中文网/ 问答频道 /正文

我有一个列表,例如:

[[1, 2, 3],
 [4, 5, 6],
 [7, 8, 9]]

我需要得到这个:

[[1, 4, 7],
 [2, 5, 8],
 [3, 6, 9]]

我还需要得到对角线。 我最好在不导入其他库的情况下执行此操作


Tags: 列表情况对角线
1条回答
网友
1楼 · 发布于 2024-03-28 09:44:24
arr = [[1, 2, 3],
       [4, 5, 6],
       [7, 8, 9]]

[list(i) for i in zip(*arr)]

输出:

[[1, 4, 7], [2, 5, 8], [3, 6, 9]]

要获得对角线,请执行以下操作:

[arr[x][x] for x in range(len(arr))]

输出:

[1, 5, 9]

相关问题 更多 >