python - Python 3 中嵌套列表的对角线

Question

我在一个列表中有几个列表，它们形成了一个圈和叉的游戏:

list = [
        ['X', 'O', 'O'], 
        ['O', 'X', 'O'], 
        [' ', 'X', ' ']
       ]

我需要编写一个函数，返回游戏的对角线从左上角到右下角，然后从右上角到左下角，因此输出为:

diags = (['X', 'X', ' '],['O', 'X', ' '])

我尝试过各种嵌套 for 循环的组合，但似乎无法理解它。

Answer 1

nw_to_se = [your_list[i][i] for i in range(3)]
ne_to_sw = [your_list[i][2-i] for i in range(3)]
diags = (nw_to_se, ne_to_sw)

您还可以使用 [-i-1] 来代替 [2-i]，它可以缩放到任意大小的正方形。