awp 发表于 2023-7-29 12:23:26

关于yield和return

def gen():
    print('Loop1')
    yield 100
    print('Loop2')
    return 200


for i in gen():
    print(i)

运行结果是:
Loop1
100
Loop2

求问高手们:
(不必讲yield和return的区别)

return的200去哪了
在这段代码上加什么语句,才能把这个200显示出来。

陶远航 发表于 2023-7-29 12:23:36

在这段代码中,return语句的作用是结束生成器函数并返回一个值。然而,在for循环中使用生成器函数时,它不会直接返回这个值,而是通过yield语句逐个产生值。

要显示返回值200,可以使用StopIteration异常来捕获生成器迭代的结束,然后从异常对象中获取返回值。以下是修改后的代码示例:

def gen():
    print('Loop1')
    yield 100
    print('Loop2')
    return 200

g = gen()
for i in g:
    print(i)

try:
    result = next(g)
except StopIteration as e:
    result = e.value

print(result)


输出结果将为:

Loop1
100
Loop2
200


通过使用try-except块捕获StopIteration异常,并从异常对象e中获取返回值,我们可以将200显示出来。
如果问题已经解决,请设置最佳答案

isdkz 发表于 2023-7-29 12:23:36

在这段代码中, return 语句虽然存在于生成器函数 gen() 中,但是由于生成器函数的特性,当使用 for i in gen() 迭代生成器时,生成器函数的执行流程在执行到 return 语句时会直接终止,导致 return 后面的值(即200)无法被获取到或显示出来。在生成器函数中, return 可以被认为是一种终止符,用于结束生成器的迭代。

要显示出 return 后面的值(即200),可以通过捕获生成器的 StopIteration 异常来实现。在 for 循环中使用 try 和 except 块来捕获异常,并在异常块中获取 return 的值。下面是修改后的代码:

def gen():
    print('Loop1')
    yield 100
    print('Loop2')
    return 200

gen_obj = gen()
for i in gen_obj:
    print(i)
try:
    print(next(gen_obj))
except StopIteration as e:
    print('return的值是:', e.value)


输出结果为:

Loop1
100
Loop2
return的值是: 200


在修改后的代码中,我们先创建了生成器对象 gen_obj ,然后使用 for 循环迭代生成器获取前面的值(即100),最后通过捕获 StopIteration 异常获取 return 后面的值(即200),从而成功显示出了 return 的值。

awp 发表于 2023-7-29 12:25:27

isdkz 发表于 2023-7-29 12:23
在这段代码中, return 语句虽然存在于生成器函数 gen() 中,但是由于生成器函数的特性,当使用 for i in g ...

大佬,你这解答疑问的速度快到有点不像人类啊。Orz太猛了

isdkz 发表于 2023-7-29 12:30:01

awp 发表于 2023-7-29 12:25
大佬,你这解答疑问的速度快到有点不像人类啊。Orz太猛了

写了个chatgpt脚本{:5_109:}

awp 发表于 2023-7-29 12:32:22

isdkz 发表于 2023-7-29 12:30
写了个chatgpt脚本

Orz
页: [1]
查看完整版本: 关于yield和return