Задать переменные в понимании списка python

Question

Если у меня есть 2 или 3 из тех же вычислений, выполненных внутри генератора для каждого цикла, есть ли способ просто установить их как переменную?

Быстрый пример будет выглядеть так:

#Normal 
[len(i) for i in list if len(i) > 1] 

#Set variable 
[x for i in list if x > 1; x = len(i)] 

Перед кем говорит len(i) будет так быстро разница будет незначительной, я имею в виду для других вычислений, используя Len только что сделал это легче читать. Также, если есть способ, как бы вы установили несколько переменных?

Извинения, если это было задано раньше, но я искал вокруг и ничего не нашел.

Answer 1

Один из способов обойти дорогостоящую операцию, чтобы гнездо генератора в списке понимание, что просто действует как фильтр, например

def foo(x):  # assume this function is expensive 
    return 2*x 

>>> [j for j in (foo(i) for i in range(6)) if j > 4] 
#    ^only called once per element 
[6, 8, 10] 

Используя аналогичные функции и переменные к вашему примеру, вы бы

[x for x in (len(i) for i in list) if x > 1] 

Answer 2

с использованием itertools.imap в python2 будет эффективным способом сделать то, что вам нужно, и, скорее всего, опережать выражение генератора:

[x for x in imap(len, lst) if x > 4] 

Answer 3

Большинство реализаций Python, как вы правильно поняли, не имеют общей оптимизации подвыражения, поэтому ваше первое выражение действительно вызовет len (x) дважды за итерацию. Итак, почему не просто два соображения:

a = [len(x) for x in list] 
b = [x for x in a if x > 1] 

Это делает два прохода, но только один вызов len Если функция была дорогой, это, вероятно, победа. Я должен был это время убедиться.

Вложенная версия Cyber ​​- это по сути то же самое.