生成器表达式中的多个产量？ Python

Question

我一直在尝试将此生成器函数转换为生成器表达式，但没有成功。生成器表达式是否可以具有相同的输出？

    def gen5(num):
         for i in range(num):             
             yield i *2
             yield i *10
             yield i *i

    g = gen5(4) 
    list(g)
    [0, 0, 0, 2, 10, 1, 4, 20, 4, 6, 30, 9]

我尝试了以下方法，但得到了这个 (None, None, None) 输出。

    gen_exp2 = (((yield u*2), (yield u*10) ,(yield u*u)) for u in  range(4))

    list(gen_exp2)

    [0,
     0,
     0,
     (None, None, None),
     2,
     10,
     1,
    (None, None, None),
     4,
    20,
     4,
    (None, None, None),
    6,
    30,
    9,
   (None, None, None)]

我也这样做了，但它提供了 4 个嵌套元组，而不是上面的列表。

       gen_exp3 = (((i*2), (i*10), (i*i)) for i in range(4))

       list(gen_exp3)

       [(0, 0, 0), (2, 10, 1), (4, 20, 4), (6, 30, 9)]

另外，如何向生成器表达式添加参数？ 非常感谢。

Answer 1

你只需要在 gen exp 中有一个双循环：

num = 4
g = (j for i in range(num) for j in (i*2, i*10, i*i))
print(*g)

输出

0 0 0 2 10 1 4 20 4 6 30 9

正如 Moses 所说，在 gen exp 中使用 yield 不是一个好主意。

---

这是一个可爱的变体，使用map

g = (j for i in range(num) for j in map(i.__mul__, (2, 10, i)))

但是，有些人可能不喜欢 i.__mul__ 的这种用法。

---

您问：“如何向生成器表达式添加参数？”。好吧，你不能。你可以创建一个lambda，正如 Moses 所示，但实际上你最好创建一个适当的生成器函数。 Lambda 应该是匿名函数，因此创建 lambda 并将其绑定到名称是一种相当可疑的做法。 ;)

Answer 2

您不需要在生成器表达式中使用yield。

请参阅yield in list comprehensions and generator expressions，了解为什么这很容易被视为错误，尽管事实并非如此。

您可以使用嵌套推导在生成器表达式中实现相同的目的：

gen_exp2 = (x for u in range(4) for x in (u*2, u*10, u*u))

您可以通过将生成器表达式放入函数中来对其进行参数化：

gen_fun = lambda num: (x for u in range(num) for x in (u*2, u*10, u*u)) 

不过，保留原来的方法可能会更好，因为在函数中使用 yield 可能比在生成器表达式中使用嵌套推导更快，而生成器表达式在没有嵌套的情况下已经很慢了。