ایتریتور (یکی یکی پیش رفتن) در پایتون

ایتریتور یک شیء با مجموعه‌ای قابل شمارش از مقادیر است. می‌توانیم به مقادیر آن یکی یکی دسترسی داشته باشیم. در پایتون، یک ایتریتور کلاسی است که از دو متد __iter__() و __next__() پیروی می‌کند.

تفاوت ایتریتور و آیترابل (Iterable)

مجموعه‌هایی مثل list (لیست)، tuple (تاپل)، dictionary (دیکشنری) و set (مجموعه) همه iterable هستند، یعنی می‌توانیم از آنها یک iterator بسازیم. مهم‌ترین نکته اینجاست که همه iterableها متد iter() را دارند که یک iterator تولید می‌کند.

مثال: گرفتن `iterator` از `tuple`

mytuple = ("apple", "banana", "cherry")
myit = iter(mytuple)

print(next(myit)) # apple
print(next(myit)) # banana
print(next(myit)) # cherry

مثال: گرفتن iterator از string

mystr = "banana"
myit = iter(mystr)

print(next(myit)) # b
print(next(myit)) # a
print(next(myit)) # n

مرور بر ایتریتور با `for loop`

به‌جای استفاده از next() می‌توانیم از یک حلقه for برای مرور روی iterator استفاده کنیم.

مثال: مرور بر `tuple`

mytuple = ("apple", "banana", "cherry")
for x in mytuple:
  print(x)

یک ایتریتور خودمان بسازیم!

برای ساخت یک iterator خودمان، باید دو متد __iter__() و __next__() را در یک کلاس تعریف کنیم.

مثال: ایتریتور شمارنده اعداد

class MyNumbers:
    def __iter__(self):
        self.a = 1
        return self

    def __next__(self):
        x = self.a
        self.a += 1
        return x

myclass = MyNumbers()
myiter = iter(myclass)

print(next(myiter)) # 1
print(next(myiter)) # 2
print(next(myiter)) # 3

متوقف کردن یک ایتریتور

اگر بخواهیم پس از تعداد مشخصی مقدار StopIteration رخ دهد، می‌توانیم در __next__() این مورد را کنترل کنیم.

مثال: متوقف کردن ایتریتور پس از 5 مقدار

class MyNumbers:
    def __iter__(self):
        self.a = 1
        return self

    def __next__(self):
        if self.a > 5:
            raise StopIteration
        x = self.a
        self.a += 1
        return x

myclass = MyNumbers()
myiter = iter(myclass)

for x in myiter:
    print(x)  # 1 2 3 4 5

جمع‌بندی

iterable ها شامل list، tuple، dict و set هستند.
iterator یک شیء است که دارای __iter__() و __next__() است.
for loop به‌طور خودکار از iterator استفاده می‌کند.
می‌توانیم iterator خودمان را با یک کلاس سفارشی‌سازی کنیم.
با StopIteration می‌توانیم مقداردهی را متوقف کنیم.

آموزش رایگان صفر تا صد پایتون (Python) به زبان ساده