python列表的嵌套该怎么理解

Question

>>> matrix = [
...     [1, 2, 3, 4],
...     [5, 6, 7, 8],
...     [9, 10, 11, 12],
... ]

>>> [[row[i] for row in matrix] for i in range(4)]
[[1, 5, 9], [2, 6, 10], [3, 7, 11], [4, 8, 12]]

首先[row[i] for row in matrix]得到的如[1,2,3,4]的这种列表元素对吗，但之后与
for i in range(4) 又是如何进一步组合的呢？

Answer 1

不对…

这里

[row[i] for row in matrix] for i in range(4)

是一个整体，不然row[i] for row in matrix里的i是哪来的…

这个List Comprehension相当于一个嵌套循环，外层循环变量是i，内层循环变量是row。

把列表的外层循环改为普通for循环的话，等价于：

temp_list = []
for i in range(4):
    temp_list.append([row[i] for row in matrix])

把列表的两层循环都改为普通for循环写法，大致相当于：

temp_list = []
for i in range(4):
    temp_list.append([])
    for row in matrix:
        temp_list[-1].append(row[i])