Python，从生成器产生的紧凑方法[重复]

Question

我有一个生成器，我希望得到它。

def foo():
    a = map(lambda x: x*2, range(5))
    # I want a better way to the next few lines
    # The one that looks more functional
    for i in a:
        yield i

我有地图、过滤器等。我想获得让步，有其他方法吗？我查看了itertools 和functools，找不到任何东西。

编辑：

为了更清楚，我想要一种在每次函数调用时返回一个值的方法。

Answer 1

试试yield from（python版本必须>= 3.3）；

def foo():
    a = map(lambda x: x*2, range(5))
    yield from a

或者任何版本，都可以使用iter：

def foo():
    a = map(lambda x: x*2, range(5))
    return iter(a)

iter 相当于生成器：

(i for i in seq)

完整示例：

def foo():
    a = map(lambda x: x*2, range(5))
    return (i for i in a)

更新：

a = iter(map(lambda x: x*2, range(5)))
def foo():
    return next(a)

print(foo())
print(foo())

输出：

0
2

顺便说一下，在这种情况下它是map，所以不需要iter：

a = map(lambda x: x*2, range(5))
def foo():
    return next(a)

print(foo())
print(foo())

输出：

0
2