code_snippets/Generators/square_nums.py

# def square_numbers(nums):
#     for i in nums:
#         yield (i*i)

# my_nums = square_numbers([1,2,3,4,5])

my_nums = (x*x for x in [1,2,3,4,5])

print list(my_nums) # [1, 4, 9, 16, 25]

# for num in my_nums:
#     print num