How to create a class as an iterator in Python

1 Answer

0 votes
class Test:
    def __iter__(self):
        self.n = 1
        return self

    def __next__(self):
        val = self.n
        self.n += 1
        return val

o = Test()
it = iter(o)

print(next(it))
print(next(it))
print(next(it))
print(next(it))
print(next(it))
print(next(it))
print(next(it))


'''
run:

1
2
3
4
5
6
7

'''

 



answered Nov 24, 2018 by avibootz

Related questions

...