Есть ли ограничение на количество циклов for, которые вы можете вкладывать в понимание списка? - PullRequest
Новая
       0

Есть ли ограничение на количество циклов for, которые вы можете вкладывать в понимание списка?

1 голос
/ 20 апреля 2020

Мне было интересно, почему следующее понимание списка не удалось (UnboundLocalError: local variable 'y_value' referenced before assignment). Если я буквально произношу циклы for в тройной вложенности для l oop, я получаю нужные результаты. 

for y_value_group in collinear_y_values:
    all_y_values = []
    for y_value in y_value_group:
        for line_id in horizontal_lines[y_value]:
            for p in LINES[line_id].pts:
                all_y_values.append(p.y)
    print(all_y_values)
    all_y_values = [p.y for p in LINES[line_id].pts for line_id in horizontal_lines[y_value] for y_value in y_value_group]
    print(all_y_values)

дает следующий вывод: 

[-0.01447138307529966, 0.22089181280929138, 0.22089181280929138, 0.19409634248390767]
---------------------------------------------------------------------------
UnboundLocalError                         Traceback (most recent call last)
<ipython-input-85-62411ee08ee6> in <module>
     24     print(all_y_values)
---> 25     all_y_values = [p.y for p in LINES[line_id].pts for line_id in horizontal_lines[y_value] for y_value in y_value_group]
     26     print(all_y_values)

<ipython-input-85-62411ee08ee6> in <listcomp>(.0)
     24     print(all_y_values)
---> 25     all_y_values = [p.y for p in LINES[line_id].pts for line_id in horizontal_lines[y_value] for y_value in y_value_group]
     26     print(all_y_values)

UnboundLocalError: local variable 'y_value' referenced before assignment

1 Ответ

5 голосов
/ 20 апреля 2020

Несколько запутанно, в понимании списка вы должны сначала поставить внешнюю l oop: 

>>> [x for x in y for y in np.random.random((2,2))]
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
NameError: name 'y' is not defined
>>> [x for y in np.random.random((2,2)) for x in y]
[0.5656047153549479, 0.19139220091114273, 0.10286775868807774, 0.3230695608882298]

Так что просто измените порядок: 

[
    p.y 
    for y_value in y_value_group
    for line_id in horizontal_lines[y_value]
    for p in LINES[line_id].pts
]
...