Python 迭代器的设置和使用方法

• 2020 年 4 月 19 日
• AI
• Python

一、理解迭代器

一般情况下，迭代器可以通过一个 for 循环实现。
需要有两个要素：

1. 有一个容器，放置可迭代的数据。
2. 每次从容器中取得一个数据，实现迭代。

二、代码实现

比如利用列表作为迭代器的容器：

a = [1,3,6,8]
d = 0

for i in a:
    d+=i

print(d)

比如利用 range() ：

e = 0

for j in range(5):
    e+=j

print(e)

三、利用 python 的内置函数

Python 中有两个和迭代相关的内置函数：

1. iter()，对应的是魔法方法 __iter__()
2. next()，对应的是魔法方法 __next__()

f = 'AI'
item = iter(f)
print(next(item))
print(next(item))
print(next(item))

# 显示结果
A
I
StopIteration

也可以加入循环语句使用：

g = [1,3,6,8]
item = iter(g)
while True:
    try:
        each = next(item)
    except StopIteration:
        break
    print(each)

# 显示结果
1
3
6
8