Why Dictionary comprehension with condition in Python? - Purpose & Use Cases

The Big Idea

What if you could filter data instantly with just one simple line of code?

The Scenario

Imagine you have a dictionary of items with prices, and you want to create a new dictionary that only includes items costing less than $20. Doing this by hand means checking each item one by one and writing down only the cheap ones.

The Problem

Manually filtering items is slow and tiring, especially if the list is long. It's easy to make mistakes by missing items or including wrong ones. Also, writing repetitive code for this task can be boring and error-prone.

The Solution

Dictionary comprehension with a condition lets you quickly create a new dictionary by picking only the items that meet your rule, all in one neat line of code. This saves time, reduces mistakes, and makes your code clean and easy to read.

Before vs After

✗ Before

new_dict = {}
for k, v in old_dict.items():
    if v < 20:
        new_dict[k] = v

✓ After

new_dict = {k: v for k, v in old_dict.items() if v < 20}

What It Enables

This lets you quickly filter and transform data, making your programs smarter and more efficient with less effort.

Real Life Example

Suppose you run a store and want to send a discount email only to customers who spent more than $100. Using dictionary comprehension with condition, you can easily create a dictionary of those customers without extra loops.

Key Takeaways

Manual filtering is slow and error-prone.

Dictionary comprehension with condition simplifies filtering in one line.

It makes your code cleaner, faster, and easier to maintain.

Practice

1.

What does the following dictionary comprehension do?
{k: v for k, v in {'a': 1, 'b': 2, 'c': 3}.items() if v > 1}

easy

A. {'a': 1, 'b': 2, 'c': 3} - keeps all items

B. {'a': 1} - keeps items with values equal to 1

C. {'b': 2, 'c': 3} - keeps items with values greater than 1

D. {} - removes all items

Solution

1. Step 1: Understand the dictionary comprehension structure

   The comprehension loops over each key-value pair in the original dictionary.

2. Step 2: Apply the condition if v > 1

   Only pairs where the value is greater than 1 are included in the new dictionary.

3. Final Answer:

   {'b': 2, 'c': 3} - keeps items with values greater than 1 -> Option C

4. Quick Check:

   Filter values > 1 = {'b': 2, 'c': 3} [OK]

Hint: Look for the condition after the for loop to filter items [OK]

Common Mistakes:

• Ignoring the condition and including all items
• Confusing keys and values in the condition
• Using wrong comparison operator

2.

Which of the following is the correct syntax for a dictionary comprehension with a condition?

?

easy

A. {k: v for k, v in d.items() if v % 2 == 0}

B. {k: v if v % 2 == 0 for k, v in d.items()}

C. {k: v for k, v in d.items() where v % 2 == 0}

D. {k: v for k, v in d.items() when v % 2 == 0}

Solution

1. Step 1: Recall dictionary comprehension syntax

   The correct syntax is {key: value for key, value in iterable if condition}.

2. Step 2: Identify the correct option

   {k: v for k, v in d.items() if v % 2 == 0} matches the correct syntax with if after the loop. Others use invalid keywords or wrong order.

3. Final Answer:

   {k: v for k, v in d.items() if v % 2 == 0} -> Option A

4. Quick Check:

   Correct syntax uses 'if' after for loop [OK]

Hint: Remember: condition goes after the for clause in comprehension [OK]

Common Mistakes:

• Placing 'if' before the for loop
• Using 'where' or 'when' instead of 'if'
• Incorrect order of clauses

3.

What is the output of this code?

nums = {'x': 10, 'y': 5, 'z': 0}
filtered = {k: v for k, v in nums.items() if v}
print(filtered)

medium

A. {'x': 10, 'y': 5}

B. {}

C. {'z': 0}

D. {'x': 10, 'y': 5, 'z': 0}

Solution

1. Step 1: Understand the condition if v

   This condition filters out values that are 'falsy' in Python, like 0, None, or empty.

2. Step 2: Apply the condition to each item

   Values 10 and 5 are truthy, so included; 0 is falsy, so excluded.

3. Final Answer:

   {'x': 10, 'y': 5} -> Option A

4. Quick Check:

   Falsy values excluded = {'x': 10, 'y': 5} [OK]

Hint: Falsy values like 0 are skipped when condition is just 'if v' [OK]

Common Mistakes:

• Including zero as truthy
• Confusing keys and values
• Expecting all items to be included

4.

Find the error in this dictionary comprehension:

data = {'a': 1, 'b': 2, 'c': 3}
result = {k: v for k, v in data.items() if v > 1 else 0}
print(result)

medium

A. No error, outputs {'b': 2, 'c': 3}

B. SyntaxError due to incorrect use of else in comprehension

C. Outputs {'a': 0, 'b': 2, 'c': 3}

D. KeyError because of else clause

Solution

1. Step 1: Check the use of else in comprehension condition

   Dictionary comprehensions cannot have an else clause directly after the if condition.

2. Step 2: Identify correct syntax

   To use else, it must be inside a value expression with a ternary operator, not after if.

3. Final Answer:

   SyntaxError due to incorrect use of else in comprehension -> Option B

4. Quick Check:

   else must be inside value expression, not after if [OK]

Hint: Use ternary inside value, not else after if in comprehension [OK]

Common Mistakes:

• Placing else after if in comprehension
• Confusing ternary operator syntax
• Expecting else to filter keys

5.

You have a dictionary of student scores:
scores = {'Alice': 85, 'Bob': 42, 'Charlie': 73, 'David': 58}
Use dictionary comprehension with a condition to create a new dictionary passed containing only students who scored 60 or more, but store their scores as 'Pass' instead of the number.
Which code correctly does this?

hard

A. {k: 'Pass' for k, v in scores.items() if v > 60}

B. {k: 'Pass' if v >= 60 else 'Fail' for k, v in scores.items()}

C. {k: v for k, v in scores.items() if v >= 60}

D. {k: 'Pass' for k, v in scores.items() if v >= 60}

Solution

1. Step 1: Understand the requirement

   We want only students with scores 60 or more, and their values replaced by 'Pass'.

2. Step 2: Check each option

   {k: 'Pass' for k, v in scores.items() if v >= 60} filters with if v >= 60 and sets value to 'Pass'. {k: 'Pass' if v >= 60 else 'Fail' for k, v in scores.items()} includes all students with ternary but no filtering. {k: v for k, v in scores.items() if v >= 60} keeps original scores. {k: 'Pass' for k, v in scores.items() if v > 60} excludes 60 exactly.

3. Final Answer:

   {k: 'Pass' for k, v in scores.items() if v >= 60} -> Option D

4. Quick Check:

   Filter and replace value with 'Pass' = {k: 'Pass' for k, v in scores.items() if v >= 60} [OK]

Hint: Filter with if, set value directly to 'Pass' in comprehension [OK]

Common Mistakes:

• Using else without filtering
• Not including scores equal to 60
• Keeping original scores instead of 'Pass'