Python iter()用法及代碼示例

iter() 函數創建一個可以一次迭代一個元素的對象。

這些對象在與 for loop 、 while loop 等循環結合使用時很有用。

用法:

iter(object, sentinel)

參數：

iter() 函數有兩個參數：

• object- 必須創建其迭代器的對象(可以是套,元組， 等等。)
• 哨兵(可選)- 用於表示序列結束的特殊值

返回：

• iter() 函數返回給定對象的迭代器對象。
• 如果用戶定義的對象沒有實現 __iter__() 和 __next__() 或 __getitem()__ ，則會引發 TypeError 異常。
• 如果還提供了 sentinel 參數，則 iter() 返回一個迭代器，直到找不到 sentinel 字符。

示例 1：Python iter() 的工作

# list of vowels
vowels = ['a', 'e', 'i', 'o', 'u']
vowels_iter = iter(vowels)

print(next(vowels_iter))    # 'a'
print(next(vowels_iter))    # 'e'
print(next(vowels_iter))    # 'i'
print(next(vowels_iter))    # 'o'
print(next(vowels_iter))    # 'u'

輸出

a
e
i
o
u

示例 2：iter() 用於自定義對象

class PrintNumber:
    def __init__(self, max):
        self.max = max

    def __iter__(self):
        self.num = 0
        return self

    def __next__(self):
        if(self.num >= self.max):
            raise StopIteration
        self.num += 1
        return self.num

print_num = PrintNumber(3)

print_num_iter = iter(print_num)
print(next(print_num_iter))  # 1
print(next(print_num_iter))  # 2
print(next(print_num_iter))  # 3

# raises StopIteration
print(next(print_num_iter))

輸出

1
2
3
Traceback (most recent call last):
  File "", line 23, in 
File "", line 11, in __next__
StopIteration

示例 3：iter() 帶哨兵參數

class DoubleIt:

    def __init__(self):
        self.start = 1

    def __iter__(self):
        return self

    def __next__(self):
        self.start *= 2
        return self.start

    __call__ = __next__
    
my_iter = iter(DoubleIt(), 16)

for x in my_iter:
    print(x)

輸出

2
4
8

在這裏，我們實現了一個沒有StopIteration 條件的自定義可迭代對象。

但是，我們可以使用帶有sentinel 參數的iter() 方法來停止迭代。如果從__next__() 返回的值等於sentinel , StopIteration 將被提升，否則將返回該值。