イテレータの実装について，スキップしたい場合

Question

イテレータの一部をスキップしたい

1, 2, 3, 4, .. とカウントアップしていくイテレータを少しいじって，例えば，3を含む数字はスキップしたい場合，どう実装したらいいのか教えていただけますでしょうか？

イテレータの実装例

https://qiita.com/propella/items/32cf705a6eeb42a04d86より引用

class Counter():

    def __init__(self, max_):
        self.count = 0
        self.max_ = max_

    def __iter__(self):
        return self

    def __next__(self):
        self.count += 1
        if self.count <= self.max_:
            return self.count
        raise StopIteration()

の，def __next__(self)の部分だと思いますが，

    def __next__(self):
        self.count += 1
        # ここに何らかの処理？
        # if self.count.has_3():
        #     skip .. ??
        if self.count <= self.max_:
            return self.count
        raise StopIteration()

みたいなことをやりたいのですが，どうしたら良いでしょうか．
よろしくお願いいたします．

Answer 1

whileでぶん回すのが楽そうです。

def __next__(self):
    self.count += 1
    while '3' in f'{self.count}':
        self.count += 1
        
    if self.count <= self.max_:
        return self.count

    raise StopIteration()

Comments

[macg3n]

たしかに，，シンプルでいいですね！ありがとうございます．

Answer 2

条件次第ではありかなというコードです。。。

    def __next__(self):
        self.count += 1
        if self.count.has_3():  # 判定処理は上手いこと直しましょう
            return self.__next__()

        if self.count <= self.max_:
            return self.count
        raise StopIteration()

---

難点：30000から1万回再帰とかすると、再帰回数上限に引っかかる

sys.setrecursionlimitをいじるのも相当バカバカしいですし。

Comments

[macg3n]

なるほど！！そういう使い方もあるのですね．勉強になります．ありがとうございます