FilesExpand file tree

master

/

iterator.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

39 lines (28 loc) · 952 Bytes

master

/

iterator.py

Copy path

Top

File metadata and controls

• Code
• Blame

39 lines (28 loc) · 952 Bytes

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

#!/usr/bin/env python

# -*- coding: utf-8 -*-

"""http://ginstrom.com/scribbles/2007/10/08/design-patterns-python-style/

Implementation of the iterator pattern with a generator"""

from __future__ import print_function

def count_to(count):

"""Counts by word numbers, up to a maximum of five"""

numbers = ["one", "two", "three", "four", "five"]

# enumerate() returns a tuple containing a count (from start which

# defaults to 0) and the values obtained from iterating over sequence

for pos, number in zip(range(count), numbers):

yield number

# Test the generator

count_to_two = lambda: count_to(2)

count_to_five = lambda: count_to(5)

print('Counting to two...')

for number in count_to_two():

print(number, end=' ')

print()

print('Counting to five...')

for number in count_to_five():

print(number, end=' ')

print()

### OUTPUT ###

# Counting to two...

# one two

# Counting to five...

# one two three four five