为什么生成器上的zip只返回一个项目?

时间:2014-03-12 04:17:42

标签: python python-2.7 generator

我在Python中试过这个,认为它会给我[(1,123),(2,123)]:

Python 2.7.3 (default, Feb 27 2014, 19:58:35)
>>> def my_generator():
...   yield 123
...
>>> zip([1,2], my_generator())
[(1, 123)]

为什么在创建一个项目后zip会停止?是否有一种Pythonic方式来获得我想要的东西?

1 个答案:

答案 0 :(得分:7)

创建一个无限的生成器,就像这样

def my_generator():
    while True:
        yield 123
print zip([1,2], my_generator())
# [(1, 123), (2, 123)]

更好的方法是使用itertools.repeat,就像这样

from itertools import repeat
print zip([1,2], repeat(123))
# [(1, 123), (2, 123)]