python iterator and generator-优快云博客

本文链接：https://blog.youkuaiyun.com/Neil_001/article/details/103545104

本文深入探讨了Python中迭代器(iterator)与生成器(generator)的概念及应用，通过实例展示了如何使用__iter__和__next__方法实现自定义迭代器，以及如何利用yield关键字创建生成器，提供了一种高效内存友好的数据处理方式。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

1. 描述

iterator: 适用于 list, array, range

class: __iter__, __next__, 适用于类实例化后的对象

generator: 函数调用对象进行迭代 it = iter(func())

2. 代码

# -*- coding: utf-8 -*-
import sys


my_list = range(20)
my_it = iter(my_list)
print("1: ", next(my_it))
print("2: ", next(my_it))

for item in my_it:
    print(item, end=' ')


print()
print('= = ' * 30)


class IterNum:
    def __iter__(self):
        self.cnt = 1
        return self

    def __next__(self):
        if self.cnt < 10:
            cnt = self.cnt
            self.cnt = cnt + 1
            return cnt
        else:
            raise StopIteration


my_num = IterNum()
my_num_iter = iter(my_num)
while True:
    try:
        print(next(my_num_iter), end=' ')
    except StopIteration:
        print('stop...')
        break


print('* * ' * 30)


def get_next():
    n = 0
    while True:
        if n < 15:
            yield n
            n += 1
        else:
            return


func = get_next()
while True:
    try:
        print(next(func), end=' ')
    except StopIteration:
        print('stop...')
        break

3. 输出
1: 0
2: 1
2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19
= = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = =
1 2 3 4 5 6 7 8 9 stop...
* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * *
0 1 2 3 4 5 6 7 8 9 10 11 12 13 14 stop...

Process finished with exit code 0