Python基于yield遍历多个可迭代对象

使用itertools工具类中的chain方法,可以很方便的将多个iterable对象一起遍历. 不过,对于dict类型的iterable对象,只会遍历key. from itertools import chain my_list = [1, 2, 3] my_dict = { 'name': 'zs', 'age': 45 } # 使用chain将三个可迭代对象一起遍历, dict是打印key for value in chain(my_list, my_dict, range(20, 30)): print(value, end=",") # 1,2,3,name,age, 在Python编程中，遍历多个可迭代对象是一个常见的需求，特别是在处理数据集合或者组合不同数据源的时候。`itertools` 是Python内置的一个模块，提供了许多高效、低级的迭代器函数，其中 `chain()` 函数就是用来将多个可迭代对象连接成一个单一的迭代器。在本文中，我们将深入探讨如何使用 `chain()` 函数以及如何自定义类似功能的生成器，特别是涉及到 `yield` 关键字的使用。 `chain()` 函数的工作原理是将传入的可迭代对象依次连接起来，形成一个新的迭代器。在给定的例子中，`my_list`、`my_dict` 和 `range(20, 30)` 分别被作为参数传递给 `chain()`。`my_dict` 被视为可迭代对象，但由于字典的迭代默认只返回键，因此在遍历时只会看到 `'name'` 和 `'age'`。 ```python from itertools import chain my_list = [1, 2, 3] my_dict = {'name': 'zs', 'age': 45} for value in chain(my_list, my_dict, range(20, 30)): print(value, end=",") ``` 输出结果为：`1,2,3,name,age,20,21,22,23,24,25,26,27,28,29,` 然而，如果你希望同时遍历字典的键和值，你需要先将字典转换为元组的列表，然后再传递给 `chain()`。例如： ```python my_dict_items = [(k, v) for k, v in my_dict.items()] for value in chain(my_list, my_dict_items, range(20, 30)): print(value, end=",") ``` 现在，我们来讨论 `yield` 关键字和自定义生成器。`yield` 是Python中的一个特殊语句，它允许函数暂停执行，并返回一个值。当调用这个函数时，它不会立即执行，而是返回一个生成器对象。每次对生成器对象进行迭代时，函数会从上次暂停的地方继续执行，直到遇到下一个 `yield` 语句。 在例子中，我们创建了一个名为 `my_chain` 的自定义生成器函数，它接受任意数量的可迭代对象作为参数。通过 `for` 循环，我们依次遍历这些可迭代对象，并在每个内部循环中使用 `yield` 语句返回当前的值。这样，`my_chain` 函数就能像 `chain()` 一样工作了： ```python def my_chain(*args): for iterable_obj in args: for value in iterable_obj: yield value for value in my_chain(my_list, my_dict, range(20, 30)): print(value, end=",") ``` 这将输出与使用 `chain()` 相同的结果。 `yield` 在处理大量数据或实现协程（coroutines）时特别有用，因为它可以避免一次性加载所有数据到内存中，而是按需生成数据，从而节省内存资源。此外，通过使用生成器，我们可以构建更复杂的迭代逻辑，比如在遍历过程中过滤、映射或组合数据。 总结来说，`itertools.chain()` 是一个方便的工具，用于将多个可迭代对象合并为一个迭代器，而 `yield` 关键字则是实现生成器的核心，允许我们创建可暂停的函数，以实现迭代和协程等功能。通过自定义 `my_chain` 生成器，我们可以理解 `yield` 的工作原理及其在组合多个可迭代对象时的应用。