Files
code_snippets/Python/Iterators-Coding-Problem/iter-demo.py
T
2018-10-23 17:44:00 -06:00

42 lines
739 B
Python

class Sentence:
def __init__(self, sentence):
self.sentence = sentence
self.index = 0
self.words = self.sentence.split()
def __iter__(self):
return self
def __next__(self):
if self.index >= len(self.words):
raise StopIteration
index = self.index
self.index += 1
return self.words[index]
def sentence(sentence):
for word in sentence.split():
yield word
my_sentence = sentence('This is a test')
# for word in my_sentence:
# print(word)
print(next(my_sentence))
print(next(my_sentence))
print(next(my_sentence))
print(next(my_sentence))
print(next(my_sentence))
# This should have the following output:
# This
# is
# a
# test