map.py

def square(x):
    return x**2

itr1 = range(5)
for i in itr1:
    print(i)
'''
Prints
0
1
2
3
4 
'''
#Using map
itr2 = map(square, itr1)
for i in itr2:
    print(i)
'''
Prints
0
1
4
9
16 
'''