Some links that might be helpful for this story.

itertools

List comprehensions