List Comprehension in Python

4.8/5 - (6 votes)

Understanding List Comprehension

List comprehension is a concise way to create lists in Python. They offer a shorter syntax to achieve the same result as using a traditional for loop and a conditional statement. List comprehensions make your code more readable and efficient by condensing multiple lines of code into a single line.

The basic syntax for a list comprehension is:

new_list = [expression for element in iterable if condition]

Here, the expression is applied to each element in the iterable (e.g., a list or a range), and the result is appended to the new_list if the optional condition is True. If the condition is not provided, all elements will be included in the new list.

Let’s look at an example. Suppose you want to create a list of squares for all even numbers between 0 and 10. Using a list comprehension, you can write:

squares = [x**2 for x in range(11) if x % 2 == 0]

This single line of code generates the list of squares, [0, 4, 16, 36, 64, 100]. It’s more concise and easier to read compared to using a traditional for loop:

squares = []
for x in range(11):
    if x % 2 == 0:
        squares.append(x**2)

You can watch my explainer video on list comprehension here:

A Simple Introduction to List Comprehension in Python

List comprehensions can also include multiple conditions and nested loops.

For example, you can create a list of all numbers divisible by both 3 and 5 between 1 and 100 with the following code:

divisible = [num for num in range(1, 101) if num % 3 == 0 and num % 5 == 0]

In this case, the resulting list will be [15, 30, 45, 60, 75, 90].

One more advanced feature of Python list comprehensions is the ability to include conditional expressions directly in the expression part, rather than just in the condition.

For example, you can create a list of “even” and “odd” strings based on a range of numbers like this:

even_odd = ["even" if x % 2 == 0 else "odd" for x in range(6)]

This code generates the list ["even", "odd", "even", "odd", "even", "odd"].

If you want to learn to write more concise Python code, check out my book: πŸ‘‡

Check out my new Python book Python One-Liners (Amazon Link).

If you like one-liners, you’ll LOVE the book. It’ll teach you everything there is to know about a single line of Python code. But it’s also an introduction to computer science, data science, machine learning, and algorithms. The universe in a single line of Python!

The book was released in 2020 with the world-class programming book publisher NoStarch Press (San Francisco).

Publisher Link: https://nostarch.com/pythononeliners

Creating New Lists

List comprehensions provide a concise way to make new lists by iterating through an existing list or other iterable object. They are more time and space-efficient than traditional for loops and offer a cleaner syntax.

A basic example of list comprehension is creating a list of even numbers:

even_numbers = [x*2 for x in range(5)] 
# Output: [0, 2, 4, 6, 8]

This creates a new list by multiplying each element within the range(5) function by 2. This compact syntax allows you to define a new list in a single line, making your code cleaner and easier to read.

You can also include a conditional statement within the list comprehension:

even_squares = [x**2 for x in range(10) if x % 2 == 0] 
# Output: [0, 4, 16, 36, 64]

This example creates a new list of even squares from 0 to 64 by using an if statement to filter out the odd numbers. List comprehensions can also be used to create lists from other iterable objects like strings, tuples, or arrays.

For example, extracting vowels from a string:

text = "List comprehensions in Python"
vowels = [c for c in text if c.lower() in 'aeiou']  
# Output: ['i', 'o', 'e', 'e', 'o', 'i', 'o', 'i', 'o']

To create a Python list of a specific size, you can use the multiplication approach within your list comprehension:

placeholder_list = [None] * 5  
# Output: [None, None, None, None, None]

This will create a list with five None elements. You can then replace them as needed, like placeholder_list[2] = 42, resulting in [None, None, 42, None, None].

Filtering and Transforming Lists

List comprehensions in Python provide a concise way to filter and transform values within an existing list.

Filtering a list involves selecting items that meet a certain condition. You can achieve this using list comprehensions by specifying a condition at the end of the expression.

For example, to create a new list containing only even numbers from an existing list, you would write:

numbers = [1, 2, 3, 4, 5, 6, 7, 8, 9]
even_numbers = [num for num in numbers if num % 2 == 0]

In this case, the condition is num % 2 == 0. The list comprehension iterates over each item in the numbers list and only includes items where the condition is true.

You can watch my video on filtering lists here:

How to Filter a List in Python?

Aside from filtering, list comprehensions can also transform items in a list. You can achieve this by altering the expression at the beginning of the list comprehension.

For example, to create a list of squares from an existing list, you can use the following code:

squares = [num ** 2 for num in numbers]

Here, the expression num ** 2 transforms each item in the list by squaring it. The squares list will now contain the squared values of the original numbers list.

By combining filtering and transformation, you can achieve even more powerful results in a single, concise statement.

For instance, to create a new list containing the squares of only the even numbers from an existing list, you can write:

even_squares = [num ** 2 for num in numbers if num % 2 == 0]

In this example, we simultaneously filter out odd numbers and square the remaining even numbers.

To further explore list comprehensions, check out these resources on

Code Optimization and Readability

List comprehensions in Python provide a way to create a new list by filtering and transforming elements of an existing list while significantly enhancing code readability. They enable you to create powerful functionality within a single line of code. Compared to traditional for loops, list comprehensions are more concise and generally preferred in terms of readability.

Here’s an example of using a list comprehension to create a list containing the squares of even numbers in a given range:

even_squares = [x ** 2 for x in range(10) if x % 2 == 0]

This single line of code replaces a multiline for loop as shown below:

even_squares = []
for x in range(10):
    if x % 2 == 0:
        even_squares.append(x ** 2)

As you can see, the list comprehension is more compact and easier to understand. In addition, it often results in improved performance. List comprehensions are also useful for tasks such as filtering elements, transforming data, and nesting loops.

Here’s another example – creating a matrix transpose using nested list comprehensions:

matrix = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
transpose = [[row[i] for row in matrix] for i in range(len(matrix[0]))]

This code snippet is equivalent to the nested for loop version:

transpose = []
for i in range(len(matrix[0])):
    row_list = []
    for row in matrix:
        row_list.append(row[i])
    transpose.append(row_list)

While using list comprehensions, be mindful of possible downsides, including loss of readability if the expression becomes too complex. To maintain code clarity, it is crucial to strike the right balance between brevity and simplicity.

List Comprehensions with Different Data Types

List comprehensions work with various data types, such as strings, tuples, dictionaries, and sets.

For example, you can use list comprehensions to perform mathematical operations on list elements. Given a list of integers, you can easily square each element using a single line of code:

num_list = [2, 4, 6]
squared_list = [x**2 for x in num_list]

Handling strings is also possible with list comprehensions. When you want to create a list of the first letters of a list of words, use the following syntax:

words = ["apple", "banana", "cherry"]
first_letters = [word[0] for word in words]

Working with tuples is very similar to lists. You can extract specific elements from a list of tuples, like this:

tuple_list = [(1, 2), (3, 4), (5, 6)]
first_elements = [t[0] for t in tuple_list]

Additionally, you can use list comprehensions with dictionaries. If you have a dictionary and want to create a new one where the keys are the original keys and the values are the squared values from the original dictionary, use the following code:

input_dict = {"a": 1, "b": 2, "c": 3}
squared_dict = {key: value**2 for key, value in input_dict.items()}
Python Dictionary Comprehension - A Powerful One-Liner Tutorial

πŸ’‘ Recommended:

Lastly, list comprehensions support sets as well. When you need to create a set with the squared elements from another set, apply the following code:

input_set = {1, 2, 3, 4}
squared_set = {x**2 for x in input_set}

πŸ’‘ Recommended: Python Generator Expressions

Using Functions and Variables in List Comprehensions

List comprehensions in Python are a concise and powerful way to create new lists by iterating over existing ones. They provide a more readable alternative to using for loops and can easily add multiple values to specific keys in a dictionary.

When it comes to using functions and variables in list comprehensions, it’s important to keep the code clear and efficient. Let’s see how to incorporate functions, variables, and other elements mentioned earlier:

Using Functions in List Comprehensions You can apply a function to each item in the list using a comprehension. Here’s an example with the upper() method:

letters = ['a', 'b', 'c', 'd']
upper_letters = [x.upper() for x in letters]

This comprehension will return a new list containing the uppercase versions of each letter. Any valid function can replace x.upper() to apply different effects on the input list.

Utilizing Variables in List Comprehensions With variables, you can use them as a counter or a condition. For example, a list comprehension with a counter:

squares = [i**2 for i in range(1, 6)]

This comprehension creates a list of squared numbers from 1 to 5. The variable i is a counter that iterates through the range() function.

For a more complex example, let’s say we want to filter out odd numbers from a list using the modulo % operator:

numbers = [1, 2, 3, 4, 5, 6, 7, 8, 9]
even_numbers = [x for x in numbers if x % 2 == 0]

In this case, the variable x represents the current element being manipulated during the iteration, and it is used in the condition x % 2 == 0 to ensure we only keep even numbers.

Working with Nested List Comprehensions

Nested list comprehensions in Python are a versatile and powerful feature that allows you to create new lists by applying an expression to an existing list of lists. This is particularly useful for updating or traversing nested sequences in a concise and readable manner.

I created a video on nested list comprehensions here: πŸ‘‡

Python One-Liner Trick 9 - Nested List Comprehension

A nested list comprehension consists of a list comprehension inside another list comprehension, much like how nested loops work. It enables you to iterate over nested sequences and apply operations to each element.

For example, consider a matrix represented as a list of lists:

matrix = [
    [1, 2, 3],
    [4, 5, 6],
    [7, 8, 9]
]

To calculate the square of each element in the matrix using nested list comprehensions, you can write:

squared_matrix = [[x**2 for x in row] for row in matrix]

This code is equivalent to the following nested for loop:

squared_matrix = []
for row in matrix:
  squared_row = []
  for x in row:
    squared_row.append(x**2)
  squared_matrix.append(squared_row)

As you can see, the nested list comprehension version is much more concise and easier to read.

Python supports various sequences like lists, tuples, and dictionaries. You can use nested list comprehensions to create different data structures by combining them. For instance, you can convert the matrix above into a dictionary where keys are the original numbers and values are their squares:

matrix_dict = {x: x**2 for row in matrix for x in row}

This generates a dictionary that looks like:

{1: 1, 2: 4, 3: 9, 4: 16, 5: 25, 6: 36, 7: 49, 8: 64, 9: 81}

Advanced List Comprehension Techniques

List comprehension is a powerful feature in Python that allows you to quickly create new lists based on existing iterables. They provide a concise and efficient way of creating new lists with a few lines of code.

The first advanced technique to consider is using range() with index. By utilizing the range(len(...)) function, you can iterate over all the items in a given iterable.

numbers = [1, 2, 3, 4, 5]
squares = [number ** 2 for number in numbers]

In addition to creating new lists, you can also use conditional statements in list comprehensions for more control over the output.

For example, if you want to create a new list with only the even numbers from an existing list, you can use a condition like this:

numbers = [1, 2, 3, 4, 5, 6]
even_numbers = [num for num in numbers if num % 2 == 0]

Another useful feature is the access of elements in an iterable using their index. This method enables you to modify the output based on the position of the elements:

words = ["apple", "banana", "cherry", "date"]
capitals = [word.capitalize() if i % 2 == 0 else word for i, word in enumerate(words)]

In this example, the enumerate() function is used to get both the index (i) and the element (word). The even-indexed words are capitalized, and the others remain unchanged.

Moreover, you can combine multiple iterables using the zip() function. This technique allows you to access elements from different lists simultaneously, creating new lists based on matched pairs.

x = [1, 2, 3]
y = [4, 5, 6]
combined = [a + b for a, b in zip(x, y)]

Frequently Asked Questions

What is the syntax for list comprehensions with if-else statements?

List comprehensions allow you to build lists in a concise way. To include an if-else statement while constructing a list, use the following syntax:

new_list = [expression_if_true if condition else expression_if_false for item in iterable]

For example, if you want to create a list of numbers, where even numbers are squared and odd numbers remain unchanged:

numbers = [1, 2, 3, 4, 5]
new_list = [number ** 2 if number % 2 == 0 else number for number in numbers]

How do you create a dictionary using list comprehension?

You can create a dictionary using a dict comprehension, which is similar to a list comprehension. The syntax is:

new_dict = {key_expression: value_expression for item in iterable}

For example, creating a dictionary with square values as keys and their roots as values:

squares = {num ** 2: num for num in range(1, 6)}

How can you filter a list using list comprehensions?

Filtering a list using list comprehensions involves combining the basic syntax with a condition. The syntax is:

filtered_list = [expression for item in iterable if condition]

For example, filtering out even numbers from a given list:

numbers = [1, 2, 3, 4, 5]
even_numbers = [number for number in numbers if number % 2 == 0]

What is the method to use list comprehension with strings?

List comprehensions can be used with any iterable, including strings. To create a list of characters from a string using list comprehension:

text = "Hello, World!"
char_list = [char for char in text]

How do you combine two lists using list comprehensions?

To combine two lists using list comprehensions, use a nested loop. Here’s the syntax:

combined_list = [expression for item1 in list1 for item2 in list2]

For example, combining two lists containing names and ages:

names = ["Alice", "Bob", "Charlie"]
ages = [25, 30, 35]
combined = [f"{name} is {age} years old" for name in names for age in ages]

What are the multiple conditions in a list comprehension?

When using multiple conditions in a list comprehension, you can have multiple if statements after the expression. The syntax is:

new_list = [expression for item in iterable if condition1 if condition2]

For example, creating a list of even numbers greater than 10:

numbers = list(range(1, 20))
result = [number for number in numbers if number % 2 == 0 if number > 10]

πŸ’‘ Recommended: List Comprehension in Python β€” A Helpful Illustrated Guide

Python One-Liners Book: Master the Single Line First!

Python programmers will improve their computer science skills with these useful one-liners.

Python One-Liners

Python One-Liners will teach you how to read and write “one-liners”: concise statements of useful functionality packed into a single line of code. You’ll learn how to systematically unpack and understand any line of Python code, and write eloquent, powerfully compressed Python like an expert.

The book’s five chapters cover (1) tips and tricks, (2) regular expressions, (3) machine learning, (4) core data science topics, and (5) useful algorithms.

Detailed explanations of one-liners introduce key computer science concepts and boost your coding and analytical skills. You’ll learn about advanced Python features such as list comprehension, slicing, lambda functions, regular expressions, map and reduce functions, and slice assignments.

You’ll also learn how to:

  • Leverage data structures to solve real-world problems, like using Boolean indexing to find cities with above-average pollution
  • Use NumPy basics such as array, shape, axis, type, broadcasting, advanced indexing, slicing, sorting, searching, aggregating, and statistics
  • Calculate basic statistics of multidimensional data arrays and the K-Means algorithms for unsupervised learning
  • Create more advanced regular expressions using grouping and named groups, negative lookaheads, escaped characters, whitespaces, character sets (and negative characters sets), and greedy/nongreedy operators
  • Understand a wide range of computer science topics, including anagrams, palindromes, supersets, permutations, factorials, prime numbers, Fibonacci numbers, obfuscation, searching, and algorithmic sorting

By the end of the book, you’ll know how to write Python at its most refined, and create concise, beautiful pieces of “Python art” in merely a single line.

Get your Python One-Liners on Amazon!!