NameError in Python Nested for loops of List Compr

2019-08-30 07:23发布

Scenerio:

for i in range(6):
    for j in range(i):
        j

AFAIK, in list comprehension the right most for is the outer one so, I thought the following code will work:

[ j for j in range(i) for i in range(6)]

But to my surprise, it throws a NameError

Traceback (most recent call last):
File "<stdin>", line 1, in <module>
NameError: name 'i' is not defined

I wonder why it didn't work. Is it because python evaluates expression from Left to Right? Cause, I have resolved the issue by using parenthesis:

[ (j for j in range(i)) for i in range(6)]

which outputs a bunch of generator expressions:

[<generator object <listcomp>.<genexpr> at 0x7f3b42200d00>, <generator object <listcomp>.<genexpr> at 0x7f3b42200d58>, <generator object <listcomp>.<genexpr> at 0x7f3b42200db0>, <generator object <listcomp>.<genexpr> at 0x7f3b42200e08>, <generator object <listcomp>.<genexpr> at 0x7f3b42200e60>, <generator object <listcomp>.<genexpr> at 0x7f3b42200eb8>]

To explore what is inside these generator expressions we can simply cast them into lists i.e.

[ list(j for j in range(i )) for i in range(6)]

and the output is as expected:

[[], [0], [0, 1], [0, 1, 2], [0, 1, 2, 3], [0, 1, 2, 3, 4]]

I just want to know what is really happening here.

2条回答
迷人小祖宗
2楼-- · 2019-08-30 07:43

this code

j for j in range(i) for i in range(6) 

just like :

for j in range(i):
    for i in range(6):
        j

outer loop uses i before it is defined, so NameError occurred, i.e. your belief "the right most for is the outer one" is wrong.

You can use this code

[j for i in range(6) for j in range(i)]

and why below code work

[ (j for j in range(i)) for i in range(6)]
# parentheses make it work like this
for i in range(6):
     for j in range(i):
          j
查看更多
三岁会撩人
3楼-- · 2019-08-30 07:59

Correct, it's evaluated from left to right. To add the others' answers, I looked up the official explanation in the documentation.

List comprehensions have the form:

 [ expression for expr in sequence1
              for expr2 in sequence2 ...
              for exprN in sequenceN
              if condition ]

The for…in clauses contain the sequences to be iterated over. The sequences do not have to be the same length, because they are not iterated over in parallel, but from left to right; this is explained more clearly in the following paragraphs. The elements of the generated list will be the successive values of expression. The final if clause is optional; if present, expression is only evaluated and added to the result if condition is true.

To make the semantics very clear, a list comprehension is equivalent to the following Python code:

for expr1 in sequence1:
    for expr2 in sequence2:
    ...
        for exprN in sequenceN:
             if (condition):
                  # Append the value of
                  # the expression to the
                  # resulting list.
查看更多
登录 后发表回答