Comprehension lists in Python provide you a very concise and smart way of creating lists. In order to create a comprehension list you will use square brackets that would consist of a ‘for’ statement and it may or may not have more for or if statements.
For better understanding have a look at the following example:
numbers=[1,2,3,4,5,6,7,8,9,10] even_numbers=[x for x in numbers if x%2==0] for element in even_numbers: print(element)
Download the code Run the code
In the above example, even_number is a comprehension list. See how it is created from the numbers list.
Now , have a look at the second example. Here, we try to create list that provides squares of numbers that are given in the numbers list:
numbers=[1,2,3,4,5,6,7,8,9,10] sq_of_numbers=[x*x for x in numbers] for element in sq_of_numbers: print(element)
Download the code Run the code
Now, let’s try to produce a comprehension list that is a cross product of two lists
color=['red','blue','green'] dress=['frock', 'shirt','skirt'] color_of_dress=[(x,y) for x in color for y in dress] for element in color_of_dress: print(element)
Have a look at the output. It has two for loops. For every x in colors it looks for every dress in y and then creates the sets.