揭秘Python迭代器的实用高级用法，让你的代码焕然一新！

2023-09-04 23:46:41

Python中迭代器是一个强大的工具，它使我们能够轻松地遍历集合中的元素。在上一篇文章中，我们介绍了迭代器的一些基础知识。在本文中，我们将深入探究Python迭代器的几个高级用法，这些用法在实际场景中非常有用，可以大大简化代码的复杂度。

跳过开始部分

在实际的应用当中，比如记录的日志或者是代码等等，一般来说头部都会有没用的内容，而我们需要从真正有用的内容开始读取。

with open('file.txt', 'r') as f:
    # 跳过前5行
    for _ in range(5):
        next(f)
    # 开始读取有用内容
    for line in f:
        print(line)

生成器表达式

生成器表达式是一种非常简洁的创建迭代器的方式，它和列表推导式非常相似，但是它不会一次性创建整个列表，而是在需要的时候才生成元素。生成器表达式使用圆括号，其中的元素和列表推导式一样，都是用逗号分隔的表达式。

# 生成偶数的列表
even_numbers = (i for i in range(1, 11) if i % 2 == 0)

# 打印偶数列表
for number in even_numbers:
    print(number)

无限循环

在某些情况下，我们需要创建一个无限循环，比如在服务器端程序中。可以使用while True:语句来创建一个无限循环，但是这种方式比较丑陋。更好的方法是使用生成器函数，生成器函数是一种特殊类型的函数，它可以生成一个迭代器。生成器函数使用yield来生成元素。

# 创建一个无限循环的生成器函数
def infinite_sequence():
    i = 0
    while True:
        yield i
        i += 1

# 创建一个无限循环的迭代器
sequence = infinite_sequence()

# 打印前10个元素
for i in range(10):
    print(next(sequence))

循环控制

在某些情况下，我们需要控制循环的执行。可以使用break和continue语句来控制循环。break语句用于跳出循环，而continue语句用于跳过当前的迭代，继续执行下一个迭代。

# 使用break语句跳出循环
for i in range(1, 11):
    if i == 5:
        break
    print(i)

# 使用continue语句跳过当前的迭代
for i in range(1, 11):
    if i % 2 == 0:
        continue
    print(i)