Python 函数 - 返回生成器

Python057

Python 函数 - 返回生成器,第1张

如果函数要返回一系列结果,我们常见的方法就是将结果放到一份列表中,然后返回给调用者。比如下面的函数,返回字符串中每个单词的首字母在真个字符串中的索引:

运行结果:

上述的结果完全符合我们的预期,但 get_word_index 函数不够简洁。下面我们尝试使用生成器来实现:

运行结果:

改写之后,不仅运行结果符合要求,由于不需要和 result 列表交互,函数也变得非常简洁。下面我们就来详细学习下生成器吧~

生成器是指使用 yield 表达式的函数,调用生成器函数时,它并不会真的运行,而是会返回迭代器。每次在这个迭代器上面调用内置的 next 函数时,迭代器就会把生成器推进到下一个 yield 表达式那里。生成器传给 yield 的值均会由迭代器返回给调用者。

此外,如果输入量非常大,使用列表作为返回值,那么程序就有可能耗尽内存并崩溃。相反,使用生成器之后,则可以应对任意长度的输入数据。

例如,下面这个生成器函数可以获取文件中单词的索引,而不管文件内容多大,该函数执行时消耗的内存,只由单行的文本长度决定:

其中 test_generator.txt 中的内容如下:

运行结果:

下面这句话特别重要: 生成器函数返回的迭代器,是由状态的,及调用者不应该反复使用它 。我们那 word_index_iter 来说明:

如果想重复调用,请将其封装成容器:

运行结果:

关于上述自定义容器的实现原理,我的另外一篇文章做了详细介绍,链接奉上: https://www.jianshu.com/p/0bae644d393a

在Python中,一个函数的返回值可以直接输出,示例代码如下:

# 定义一个函数

def add(a, b):

return a + b

# 调用函数并输出返回值

print(add(1, 2))

在这段代码中,我们首先定义了一个名为add的函数,该函数接收两个参数a和b,并通过return语句返回它们的和。接着,我们调用该函数并传入参数1和2,并使用print函数输出函数的返回值,即3。

请注意,上面的代码只是一个示例,实际应用中可能需要根据实际情况进行更多的处理,比如考虑函数没有返回值等情况。

题主你好,

关键点在于函数如果没有明确使用return关键字指定函数的返回值,则默认返回值是none.

-----

所以temp = hello()这句的理解就是: 执行hello()这个函数, 并且把函数的返回值赋给变量temp, 但hello()函数中没有return语句, 所以hello()函数的返回值为默认的none.

-----

题主还可以在hello()函数最后加一个: return 123

然后再行一下看看,就明白了.

=====

希望可以帮到题主, 欢迎追问.