Learn Python in 7 Days

List comprehensions

List comprehension is a concise way of creating lists. In this section, we will use a list with the for loop. If you have not read about the for loop so far, you can skip this section and get back after learning about the for loop, covered in Chapter 6, Control Statements and Loops.

Let's take a list1 list as shown:

list1 = [2,3,4,5,6]

Now, our aim is to make a new list that contains the square of the elements of list1:

list1 = [2,3,4,5,6]list2 = []for each in list1:            list2.append(each*each)print list2

The output of the program is as follows:

Square of list

The preceding code took four lines to create the desired list. By ...