Triển khai __next__()

Phương thức __next__() là một phần quan trọng khi tạo trình lặp tùy chỉnh trong Python. Phương thức này xác định cách lấy phần tử tiếp theo trong chuỗi lần lặp. Khi triển khai __next__(), bạn kiểm soát hoạt động của trình lặp, xác định các giá trị mà nó tạo ra và thời điểm nó dừng.

Mục đích của __next__()

Phương thức __next__() có hai mục đích chính:

1. Trả về phần tử tiếp theo trong chuỗi
2. Ném ngoại lệ StopIteration khi không còn phần tử nào

Triển khai __next__()

Đây là cấu trúc cơ bản để triển khai __next__():

class MyIterator:
    def __init__(self, data):
        self.data = data
        self.index = 0

    def __iter__(self):
        return self

    def __next__(self):
        if self.index >= len(self.data):
            raise StopIteration
        result = self.data[self.index]
        self.index += 1
        return result

Các khía cạnh chính của __next__()

• Quản lý trạng thái: Theo dõi trạng thái hiện tại của lần lặp, chẳng hạn như bằng cách dùng chỉ số.
• Trả về phần tử tiếp theo: Xác định và trả về phần tử tiếp theo trong chuỗi.
• StopIteration: Ném ngoại lệ StopIteration khi không còn phần tử nào.
• Tác dụng phụ: Cập nhật mọi trạng thái nội bộ cần thiết cho lần lặp tiếp theo.

Ví dụ: trình lặp phạm vi tùy chỉnh

Đây là ví dụ về một trình lặp tùy chỉnh mô phỏng phiên bản đơn giản hóa của range()</function> trong Python:</p> <pre><code class="language-python">class CustomRange: def __init__(self, start, end): self.current = start self.end = end def __iter__(self): return self def __next__(self): if self.current >= self.end: raise StopIteration result = self.current self.current += 1 return result # Sử dụng bộ lặp phạm vi tùy chỉnh for num in CustomRange(1, 5): print(num) # Đầu ra: 1, 2, 3, 4