Skip to content
Related Articles

Related Articles

Table of Contents

Improve Article
Save Article
Like Article

Python – List Comprehension

  • Difficulty Level : Easy
  • Last Updated : 21 Oct, 2021

Python is renowned for encouraging developers and programmers to write efficient, easy-to-understand, and almost as simple-to-read code. One of the most distinctive aspects of the language is the python list and the list compression feature, which one can use within a single line of code to construct powerful functionality.

List comprehensions are used for creating new lists from other iterables like tuples, strings, arrays, lists, etc. A list comprehension consists of brackets containing the expression, which is executed for each element along with the for loop to iterate over each element. 

 Attention geek! Strengthen your foundations with the Python Programming Foundation Course and learn the basics.  

To begin with, your interview preparations Enhance your Data Structures concepts with the Python DS Course. And to begin with your Machine Learning Journey, join the Machine Learning - Basic Level Course

Syntax:



newList = [ expression(element) for element in oldList if condition ] 

Advantages of List Comprehension

  • More time-efficient and space-efficient than loops.
  • Require fewer lines of code.
  • Transforms iterative statement into a formula.

List Comprehensions vs For Loop

There are various ways to iterate through a list. However, the most common approach is to use the for loop. Let us look at the below example:

Python3




# Empty list
List = []
 
# Traditional approach of iterating
for character in 'Geeks 4 Geeks!':
    List.append(character)
 
# Display list
print(List)
Output:
[‘G’, ‘e’, ‘e’, ‘k’, ‘s’, ‘ ‘, ‘4’, ‘ ‘, ‘G’, ‘e’, ‘e’, ‘k’, ‘s’, ‘!’]
Above is the implementation of the traditional approach to iterate through a list, string, tuple, etc. Now list comprehension does the same task and also makes the program more simple. 
List Comprehensions translate the traditional iteration approach using for loop into a simple formula hence making them easy to use. Below is the approach to iterate through a list, string, tuple, etc. using list comprehension.
Python3



# Using list comprehension to iterate through loop
List = [character for character in 'Geeks 4 Geeks!']
 
# Displaying list
print(List)
Output:
[‘G’, ‘e’, ‘e’, ‘k’, ‘s’, ‘ ‘, ‘4’, ‘ ‘, ‘G’, ‘e’, ‘e’, ‘k’, ‘s’, ‘!’]


The list comprehensions are more efficient both computationally and in terms of coding space and time than a for loop. Typically, they are written in a single line of code. The below program depicts the difference between for loops and list comprehension based on performance.
Python3



# Import required module
import time
 
 
# define function to implement for loop
def for_loop(n):
    result = []
    for i in range(n):
        result.append(i**2)
    return result
 
 
# define function to implement list comprehension
def list_comprehension(n):
    return [i**2 for i in range(n)]
 
 
# Driver Code 
 
# Calculate time takens by for_loop()
begin = time.time()
for_loop(10**6)
end = time.time()
 
# Display time taken by for_loop()
print('Time taken for_loop:',round(end-begin,2))
 
# Calculate time takens by list_comprehension()
begin = time.time()
list_comprehension(10**6)
end = time.time()
 
# Display time taken by for_loop()
print('Time taken for list_comprehension:',round(end-begin,2))
Output:
Time taken for_loop: 0.56
Time taken for list_comprehension: 0.47
From the above program, we can see list comprehensions are quite faster than for loop.
Nested List Comprehensions
Nested List Comprehensions are nothing but a list comprehension within another list comprehension which is quite similar to nested for loops. Below is the program which implements nested loop:
Python3



matrix = []
 
for i in range(3):
     
    # Append an empty sublist inside the list
    matrix.append([])
     
    for j in range(5):
        matrix[i].append(j)
         
print(matrix)
Output:
[[0, 1, 2, 3, 4], [0, 1, 2, 3, 4], [0, 1, 2, 3, 4]]
Now by using nested list comprehensions same output can be generated in fewer lines of code.
Python3



# Nested list comprehension
matrix = [[j for j in range(5)] for i in range(3)]
 
print(matrix)
Output:
[[0, 1, 2, 3, 4], [0, 1, 2, 3, 4], [0, 1, 2, 3, 4]]
List Comprehensions and Lambda
Lambda Expressions are nothing but shorthand representations of Python functions. Using list comprehensions with lambda creates an efficient combination. Let us look at the below examples:


Python3



# using lambda to print table of 10
numbers = []
 
for i in range(1, 6):
    numbers.append(i*10)
 
print(numbers)
Output:
[10, 20, 30, 40, 50]
Here, we have used for loop to print table of 10.
Python3



numbers= [i*10 for i in range(1,6)]
 
print(numbers)
 Output:
[10, 20, 30, 40, 50]
 Now here, we have used only list comprehension to display table of 10.
Python3



# using lambda to print table of 10
numbers = list(map(lambda i: i*10, [i for i in range(1,6)]))
 
print(numbers)
Output:
[10, 20, 30, 40, 50]
Finally, we use lambda + list comprehension to display the table of 10. This combination is very useful to get efficient solutions in fewer lines of code for complex problems.
Conditionals in List Comprehension
We can also add conditional statements to the list comprehension. We can create a list using range(), operators, etc. and cal also apply some conditions to the list using the if statement.
Below are some examples which depict the use of list comprehensions rather than the traditional approach to iterate through iterables:
Example 1: Display square of numbers from 1 to 10.
Python3



# Getting square of even numbers from 1 to 10
squares = [n**2 for n in range(1, 11) if n%2==0]
 
# Display square of even numbers
print(squares)
Output:
[4, 16, 36, 64, 100]
Example 2: Display even elements from a list of random numbers.
Python3



# Assign matrix
twoDMatrix = [[10, 20, 30],
              [40, 50, 60],
              [70, 80, 90]]
 
# Generate transpose
trans = [[i[j] for i in twoDMatrix] for j in range(len(twoDMatrix))]
 
print(trans)
Output:
[++++[+10, 40, 70], [20, 50, 80], [30, 60, 90]]
Example 3: Toggle case of each character in a string.
Python3



# Initializing string
string = 'Geeks4Geeks'
 
# Toggle case of each character
List = list(map(lambda i: chr(ord(i)^32), string))
 
# Display list
print(List)
Output:
[‘g’, ‘E’, ‘E’, ‘K’, ‘S’, ‘\x14’, ‘g’, ‘E’, ‘E’, ‘K’, ‘S’, ‘\x01’]
Example 4: Reverse each string in a tuple.
Python3



# Reverse each string in tuple
List = [string[::-1] for string in ('Geeks', 'for', 'Geeks')]
 
# Display list
print(List)
Output:
['skeeG', 'rof', 'skeeG']
Example 5: Display the sum of digits of all the odd elements in a list.
Python3



# Explicit function
def digitSum(n):
    dsum = 0
    for ele in str(n):
        dsum += int(ele)
    return dsum
 
 
# Initializing list
List = [367, 111, 562, 945, 6726, 873]
 
# Using the function on odd elements of the list
newList = [digitSum(i) for i in List if i & 1]
 
# Displaying new list
print(newList)
Output:
[16, 3, 18, 18]
Key Points
  • Comprehension of the list is an effective means of describing and constructing lists based on current lists.
  • Generally, list comprehension is more lightweight and simpler than standard list formation functions and loops.
  • We should not write long codes for list comprehensions in order to ensure user-friendly code.
  • Every comprehension of the list can be rewritten in for loop, but in the context of list interpretation, every for loop can not be rewritten.



My Personal Notes
arrow_drop_up
Previous
Linear Equations in One Variable - Solving Equations which have Linear Expressions on one Side and Numbers on the other Side | Class 8 Maths
Next

Capgemini Interview Experience | On-Campus 2020-21
Recommended Articles
Page :
Python List Comprehension | Segregate 0's and 1's in an array list
07, Jan 18
Python List Comprehension and Slicing
10, Feb 16
Move all zeroes to end of array using List Comprehension in Python
02, Nov 17
K’th Non-repeating Character in Python using List Comprehension and OrderedDict
04, Nov 17
List comprehension and ord() in Python to remove all characters other than alphabets
08, Nov 17
Count set bits using Python List comprehension
08, Dec 17
Python List Comprehension to find pair with given sum from two arrays
23, Dec 17
Python List Comprehension | Sort even-placed elements in increasing and odd-placed in decreasing order
18, Jan 18
Python List Comprehension | Three way partitioning of an array around a given range
18, Jan 18
Difference between List comprehension and Lambda in Python
05, Jun 20
Python | List comprehension vs * operator
30, Aug 19
Python - Map vs List comprehension
21, Jan 20
How to handle a Python Exception in a List Comprehension?
20, Aug 20
Create a dictionary with list comprehension in Python
20, Nov 21
Python Dictionary Comprehension
29, Oct 17
Python | Convert list of string to list of list
22, Jan 19
Python | Convert list of tuples to list of list
26, Mar 19
Python | Convert List of String List to String List
22, Nov 19
Python | Maximum sum of elements of list in a list of lists
23, Dec 17
Sort the values of first list using second list in Python
29, Dec 17
Python | Pair and combine nested list to tuple list
01, Apr 19
Python | Filter a list based on the given list of strings
13, Jun 19
Python | Convert a nested list into a flat list
10, Sep 18
Python program to create a list of tuples from given list having number and its cube in each tuple
23, Oct 18
Article Contributed By :
https://media.geeksforgeeks.org/auth/avatar.png
riturajsaha
@riturajsaha
Vote for difficulty
Current difficulty :
Easy





Improved By :
Article Tags :
Practice Tags :
Report Issue

We use cookies to ensure you have the best browsing experience on our website. By using our site, you
acknowledge that you have read and understood our
Cookie Policy &
        Privacy Policy

Lightbox

Start Your Coding Journey Now!