How to Iterate over a list in Python using list comprehension