如何在Python中使用yield函数

Question

语法错误：'yield'在函数之外。

>>> for x in range(10):
...     yield x*x
... 
  File "<stdin>", line 2
SyntaxError: 'yield' outside function

当我尝试在for循环中使用简单的yield时，我该怎么办？

- gaurav1207

---

如果循环位于函数内部，你可以在循环中使用yield。 - vaultah

所以，如果我想在简单的for循环中使用它，我不能使用它，是否有某些原因？ - gaurav1207

可能是Python中“yield”关键字的作用是什么？的重复问题。 - iled

这是一个很好的问题，不是重复的。 - WestCoastProjects

Answer 1

编辑

根据你在评论中提到的Scala，我认为你可能需要一个列表推导式：

>>> squares = [i*i for i in range(10)]
>>> squares
[0, 1, 4, 9, 16, 25, 36, 49, 64, 81]

你也可以使用生成器表达式：

>>> squares = (i*i for i in range(10))
>>> squares
<generator object <genexpr> at 0x7f5299e04690>
>>> list(squares)
[0, 1, 4, 9, 16, 25, 36, 49, 64, 81]

你需要在函数内部调用yield。这会使函数成为生成器函数。然后，你可以迭代函数产生的连续值，例如：

def squares(N):
    for i in range(N):
        yield i*i

>>> squares(10)
<generator object squares at 0x7f5299e04500>
>>> for n in squares(10):
...    print(n)
0
1
4
9
16
25
36
49
64
81

>>> list(squares(100))
[0, 1, 4, 9, 16, 25, 36, 49, 64, 81, 100, 121, 144, 169, 196, 225, 256, 289, 324, 361, 400, 441, 484, 529, 576, 625, 676, 729, 784, 841, 900, 961, 1024, 1089, 1156, 1225, 1296, 1369, 1444, 1521, 1600, 1681, 1764, 1849, 1936, 2025, 2116, 2209, 2304, 2401, 2500, 2601, 2704, 2809, 2916, 3025, 3136, 3249, 3364, 3481, 3600, 3721, 3844, 3969, 4096, 4225, 4356, 4489, 4624, 4761, 4900, 5041, 5184, 5329, 5476, 5625, 5776, 5929, 6084, 6241, 6400, 6561, 6724, 6889, 7056, 7225, 7396, 7569, 7744, 7921, 8100, 8281, 8464, 8649, 8836, 9025, 9216, 9409, 9604, 9801]

- mhawke

---

为什么我不能在简单的for循环中使用yield，就像在Scala中一样？ - gaurav1207

Python不是Scala的简单原因。如果您想将生成的项呈现为静态结构，可以将其转换为list，如我上面所示。或者更简单地，您可以使用列表推导式：squares = [i*i for i in range(10)]。 - mhawke

@mhawke 虽然您的评论提到了这种语言中可用的途径，但实际上它确实解释了为什么不能使用简单的for循环（即为什么该语言不支持此结构）。 - WestCoastProjects

Answer 2

注意：这种语法在Python 3.7中已被弃用，并且在Python 3.8中会引发SyntaxError错误。

lamyield = lambda: [(yield x*x) for x in range(15)]
print(*lamyield()) 

另一种方式是，

lanyield = lambda: (yield from (i ** 2 for i in range(15)))
for i in lanyield():
    print(i) 

0
1
4
9
16
25
36
49
64
81
100
121
144
169
196

[Program finished]

- Subham