Python 技巧：自定义迭代器

# /*
# Python 技巧：自定义迭代器
# --------------------------------------------------------
# 功能：
# - 通过 __iter__ 与 __next__ 实现迭代对象
# */

print("示例八：自定义迭代器")

class Counter:
    def __init__(self, n):
        self.n = n
        self.i = 0

    def __iter__(self):
        return self

    def __next__(self):
        if self.i >= self.n:
            raise StopIteration
        self.i += 1
        return self.i

for x in Counter(4):
    print("迭代值:", x)

print("-" * 50)
#
# 示例八：自定义迭代器
# 迭代值: 1
# 迭代值: 2
# 迭代值: 3
# 迭代值: 4
# --------------------------------------------------