利用Python函数创建迭代器并遍历。

在Python中，迭代器是指可以遍历可迭代对象的对象。具有__iter__()和__next__()方法的对象被称为迭代器。可以使用迭代器遍历序列、列表、字典、文本文件以及许多其他对象。在本文中，我们将讨论如何在Python中创建迭代器并进行遍历。

创建迭代器

在Python中，我们可以使用关键字yield创建迭代器。 yield用于返回一个值，并保持函数的状态。 Python函数中有yield语句时，该函数会变成生成器函数，当next()函数被调用时，函数会从上次停止的地方继续执行。以下是创建迭代器的示例代码。

def sample_iterator():
   for i in range(1,6):
      yield i

my_iterator = sample_iterator()

for i in my_iterator:
   print(i)

输出结果：

1

2

3

4

5

在上面的代码中，定义了一个名为sample_iterator()的函数。该函数使用yield语句来返回数值，并保留函数的状态。然后，创建了一个名为my_iterator的迭代器对象，该迭代器是从sample_iterator()函数返回的。最后，使用for循环遍历my_iterator中的元素并打印这些元素。

Python内置模块itertools还提供了许多可以创建迭代器的函数。其中，常用的函数包括count()、cycle()和repeat()。下面我们将详细讨论这些函数的使用方法。

count()函数

count()函数用于创建一个无限的迭代器，该迭代器会从指定的数字开始并按照指定的步骤依次递增。以下是count()函数的使用示例。

import itertools

for i in itertools.count(start=5, step=2):
    print(i)
    if i > 15:
        break

输出结果：

5

7

9

11

13

15

在上述代码中，我们导入了Python内置模块itertools。然后，使用count()函数创建了一个迭代器，并定义了起始数字和步长。可以看到，迭代器从5开始并依次递增2，直到超过15。

cycle()函数

cycle()函数用于在指定序列中循环遍历元素。该函数会将序列中的所有元素复制多次，并序列化为无限的序列。以下是cycle()函数的使用示例。

import itertools

colors = ['red', 'green', 'blue']

for color in itertools.cycle(colors):
    print(color)
    if color == 'blue':
        break

输出结果：

red

green

blue

在上述代码中，我们创建了一个名为colors的列表。然后，使用cycle()函数创建一个迭代器，并循环遍历列表中的所有元素。当迭代器遍历到blue时，程序会退出循环。

repeat()函数

repeat()函数用于创建指定元素的重复序列。在repeat()函数中，可以指定元素的数量或使用无限参数，以创建无限序列。以下是repeat()函数的使用示例。

import itertools

for num in itertools.repeat(2, 5):
    print(num)

输出结果：

2

2

2

2

2

在上述代码中，我们使用repeat()函数创建了一个迭代器，生成了5个重复的数字2，并使用for循环遍历迭代器中的所有元素。

总结

在Python中，可以使用Python函数和内置模块itertools创建迭代器。关键字yield用于返回值并保持函数状态。 itertools模块中的count()、cycle()和repeat()函数用于创建特定序列的迭代器。可以通过for循环遍历迭代器，并访问其中的元素。Python中迭代器的使用和创建是Python编程的一个重要方面，学会迭代器的使用，可以使Python程序更加简洁、高效和灵活。