嵌套列表理解如何工作的解释？

Question

我理解这一点没有问题:

a = [1,2,3,4]
b = [x for x in a]

我以为就是这样,但后来我找到了这个片段:

a = [[1,2],[3,4],[5,6]]
b = [x for xs in a for x in xs]

这使得b = [1,2,3,4,5,6].问题是我在理解语法方面遇到了麻烦[x for xs in a for x in xs],有人能解释一下它是如何工作的吗？

Answer 1

啊,难以理解的"嵌套"理解.循环以与理解中相同的顺序展开.

[leaf for branch in tree for leaf in branch]

它有助于像这样思考它.

for branch in tree:
    for leaf in branch:
        yield leaf

该PEP202断言这句法与"最后的指数变化最快的"是"正确的",特别是没有的解释为什么.

Answer 2

if a = [[1,2],[3,4],[5,6]],那么如果我们展开那个列表comp,我们得到:

      +----------------a------------------+ 
      | +--xs---+ , +--xs---+ , +--xs---+ | for xs in a
      | | x , x |   | x , x |   | x , x | | for x in xs
a  =  [ [ 1 , 2 ] , [ 3 , 4 ] , [ 5 , 6 ] ]
b  =  [ x for xs in a for x in xs ] == [1,2,3,4,5,6] #a list of just the "x"s

Answer 3

b = [x for xs in a for x in xs] 类似于以下嵌套循环.

b = []
for xs in a:
   for x in xs:
       b.append(x)

Answer 4

可以这样写

result = []
for xs in a:
    for x in xs:
        result.append(x)

你可以在这里阅读更多关于它的信息

Answer 5

有效地：

...for xs in a...]

正在迭代您的主（外部）列表并依次返回您的每个子列表。

...for x in xs]

然后迭代这些子列表中的每一个。

这可以重写为：

b = []
for xs in a:
    for x in xs:
        b.append(x)

Answer 6

这是嵌套理解的一个例子。将其a = [[1,2],[3,4],[5,6]]视为 3 x 2 矩阵（矩阵 = [[1,2],[3,4],[5,6]]）。

       ______
row 1 |1 | 2 |
       ______
row 2 |3 | 4 |
       ______
row 3 |5 | 6 |
       ______

您看到的列表推导式是将矩阵中的所有元素放入列表的另一种方法。

我将尝试使用不同的变量来解释这一点，这有望更有意义。

b = [element for row in matrix for element in row]

第一个 for 循环遍历矩阵 ie 内的行[1,2],[3,4],[5,6]。第二个 for 循环遍历 2 个元素列表中的每个元素。

我在我的网站http://programmathics.com/programming/python/python-list-comprehension-tutorial/上写了一篇关于 List Comprehension 的小文章，它实际上涵盖了与这个问题非常相似的场景。我还给出了一些关于python列表理解的其他例子和解释。

免责声明：我是该网站的创建者。

Answer 7

这是我最好记住的方式：（伪代码，但具有这种类型的模式）

[(x,y,z) (loop 1) (loop 2) (loop 3)]

其中最右边的循环（循环 3）是最内层的循环。

[(x,y,z)    for x in range(3)    for y in range(3)    for z in range(3)]

其结构如下：

for x in range(3):
    for y in range(3):
        for z in range(3):
            print((x,y,z))

编辑我想添加另一种模式：

[(result) (loop 1) (loop 2) (loop 3) (condition)]

前任：

[(x,y,z)    for x in range(3)    for y in range(3)    for z in range(3)    if x == y == z]

有这种类型的结构：

for x in range(3):
    for y in range(3):
        for z in range(3):
            if x == y == z:
                print((x,y,z))