Допустим несколько значений
Можем ли мы дать более одного значения в функциях генератора питона?
Пример
In [677]: def gen():
.....: for i in range(5):
.....: yield i, i+1
.....:
In [680]: k1, k2 = gen()
---------------------------------------------------------------------------
ValueError Traceback (most recent call last)
<ipython-input-680-b21f6543a7e9> in <module>()
----> 1 k1, k2 = a()
ValueError: too many values to unpack
Это работает следующим образом:
In [678]: b = a()
In [679]: list(b)
Out[679]: [(0, 1), (1, 2), (2, 3), (3, 4), (4, 5)]
Те же результаты, даже если я это делаю:
In [692]: def a():
for i in range(5):
yield i
yield i+1
Спасибо.
Ответы
Ответ 1
Так как gen() возвращает генератор (один элемент - поэтому его нельзя распаковать как два), его нужно сначала перенести, чтобы получить значения...
g = gen()
a, b = next(g)
Он работает с list, потому что это неявно потребляет генератор.
Можно ли сделать это генератором? Что-то вроде этого:
g = gen();
def yield_g():
yield g.next();
k1,k2 = yield_g();
и, следовательно, list(k1) даст [0,1,2,3,4] и list(k2) даст [1,2,3,4,5].
Сохраните существующий генератор и используйте izip (или zip):
from itertools import izip
k1, k2 = izip(*gen())
Ответ 2
Ваша функция gen возвращает генератор, а не значения, как вы могли бы ожидать, судя по приведенному вами примеру. Если вы итерации по генератору будут получены пары значений:
In [2]: def gen():
...: for i in range(5):
...: yield i, i+1
...:
In [3]: for k1, k2 in gen():
...: print k1, k2
...:
0 1
1 2
2 3
3 4
4 5
