Introduction to Comprehensions

Comprehensions are a concise and elegant way to create lists, sets, and dictionaries in Python. They allow you to perform operations and transformations in a single line of code.

In this lesson, you’ll learn how to use list, set, and dictionary comprehensions effectively.

What Are Comprehensions?

Comprehensions are a concise way to create new collections from existing iterables. They provide a more readable and Pythonic approach compared to loops.

# Basic list comprehension
squares = [x * x for x in range(5)]
print(squares)  # Outputs: [0, 1, 4, 9, 16]

List Comprehensions

List comprehensions allow you to create lists in a single line:

# List comprehension example
numbers = [1, 2, 3, 4, 5]
doubled = [n * 2 for n in numbers]
print(doubled)  # Outputs: [2, 4, 6, 8, 10]

Set Comprehensions

Set comprehensions are similar to list comprehensions but produce sets:

# Set comprehension example
numbers = [1, 2, 2, 3, 4, 4, 5]
unique_squares = {n * n for n in numbers}
print(unique_squares)  # Outputs: {1, 4, 9, 16, 25}

Dictionary Comprehensions

Dictionary comprehensions allow you to create dictionaries in a single line:

# Dictionary comprehension example
keys = ['a', 'b', 'c']
values = [1, 2, 3]
my_dict = {k: v for k, v in zip(keys, values)}
print(my_dict)  # Outputs: {'a': 1, 'b': 2, 'c': 3}

Adding Conditions to Comprehensions

You can add conditions to comprehensions to filter values:

# Adding conditions to comprehensions
numbers = range(10)
even_numbers = [n for n in numbers if n % 2 == 0]
print(even_numbers)  # Outputs: [0, 2, 4, 6, 8]

Nesting Comprehensions

You can nest comprehensions to work with multi-dimensional data:

# Nested comprehension example
matrix = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
flattened = [item for row in matrix for item in row]
print(flattened)  # Outputs: [1, 2, 3, 4, 5, 6, 7, 8, 9]

Performance of Comprehensions

Comprehensions are faster than traditional loops because they are optimized at the C level.

Common Mistakes with Comprehensions

Here are some mistakes to avoid:

• Using comprehensions for overly complex logic.
• Creating nested comprehensions that reduce readability.
• Not considering generator expressions for large datasets.

What Did We Learn?

In this lesson, you learned:

• What comprehensions are and how they work in Python.
• How to use list, set, and dictionary comprehensions.
• How to add conditions and nest comprehensions.
• The performance benefits of using comprehensions.

Great job! Let’s move to the next topic.