Iterability in Python

Question

I am trying to understand Iterability in Python.

As I understand, __iter__() should return an object that has next() method defined which must return a value or raise StopIteration exception. Thus I wrote this class which satisfies both these conditions.

But it doesnt seem to work. What is wrong?

class Iterator:
    def __init__(self):
        self.i = 1

    def __iter__(self):
        return self

    def next(self):
        self.i += 1
        if self.i < 5:
            return self.i
        else:
            raise StopIteration

if __name__ =='__main__':
    ai = Iterator()
    b  = [i for i in ai]
    print b

Note: Updated code with incrementing i in next() and __main__ in quotes.

Answer 1

i will never become greater than 5 if you don't increment it in next()

Answer 2

Your Iterator class is correct. You just have a typo in this statement:

if __name__ ==' __main__':

There's a leading whitespace in the ' __main__' string. That's why your code is not executed at all.

Answer 3

I think in most cases it might be enough to write a generator function that uses yield instead of writing a full-fledged iterator.

Answer 4

Your current code seems to work. Instead i'll show you some more iterators/generators.

the simplest builtin with exactly your behavior.

Iterator2 = xrange(2,5)

A direct translation of your class to a generator

def Iterator3():
    i = 1
    while i < 5:
        i += 1
        yield i

a generator composed from generators in the python standard library

import itertools
Iterator4 = itertools.takewhile( 
                        lambda y : y < 5, 
                        itertools.count(2) )

a simple generator expression (not very exciting...)

Iterator5 = ( x for x in [2, 3, 4] )