Iterator pythonic way

class Trem(object):
    def __init__(self, vagoes):
        self.vagoes = vagoes

    def __iter__(self):
        for vagao in range(self.vagoes):
            yield vagao

trens = Trem(10)

for vagao in trens:
    print(vagao)