Левая часть назначения с бесконечными генераторами - PullRequest
Новая
       21

Левая часть назначения с бесконечными генераторами

0 голосов
/ 17 октября 2010

Извините, что удвоил мой предыдущий вопрос, но я подумал спросить конкретные данные, которые решат проблему. Я хочу этот результат 

tuple_of_vars = (item for _, item for zip(tuple_of_vars, new_vals_generator))

так как это невозможно 

a, b, c, d = (val for val in infite_generator)

на самом деле тогда я хочу сделать в одну строку 

for var in var_list:
    var = next(infinite_generator)

Есть ли какой-нибудь крюк интерпретатора для получения информации о количестве переменных в левой части задания? Лучше было бы, хотя, что я мог бы просто сделать автоматически этот последний бит кода (включая случаи с левой стороной, которая является слайсом с переменными индексами и шагом)

Также есть способ создать генератор переменных, который будет слева от присваивания.

РЕДАКТИРОВАТЬ: Это не останавливается в Python3: 

def incr(a):
    while True:
        yield a
        a += 1


a = [None for i in range(20)]

a[3:3:3], *_ = incr(1)

print(a)

То же самое с: 

a,b,c,d, *_ = incr(1)

print(a, b, c, d)

Даже у него нет среза (на самом деле индексы будут переменными, это всего лишь тест). Я знаю об островках и т. Д., Но это слишком медленно.

Это также приводит к ошибке: 

a = 1000*[True]

bound = int(len(a) ** 0.5)

for i in range(3, bound, 2):
    a[3::i], *x = [[False] for _ in range(bound)]

""" Error:
ValueError: attempt to assign sequence of size 1 to extended slice of size 333
"""

А это: 

a = 1000*[True]

bound = int(len(a) ** 0.5)

for i in range(3, bound, 2):
    a[3::i], *x = [False] * bound

""" Error:
TypeError: must assign iterable to extended slice
"""

1 Ответ

1 голос
/ 17 октября 2010

Когда вы знаете длину var_list, вы можете использовать itertools.islice для отключения бесконечного генератора: 

>>> import itertools
>>> infgen = itertools.cycle([1,4,9])
>>> a,b,c,d = itertools.islice(infgen, 4)
>>> a,b,c,d
(1, 4, 9, 1)

Работает и для назначения к списку. 

>>> lst = [0]*20
>>> lst[2:10:2] = itertools.islice(infgen, 4)
>>> lst
[0, 0, 4, 0, 9, 0, 1, 0, 4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]
...