Why List comprehension with if–else in Python? - Purpose & Use Cases

The Big Idea

What if you could turn long, messy loops into a single, elegant line that does it all?

The Scenario

Imagine you have a list of numbers and you want to create a new list where each number is labeled as 'Even' or 'Odd'. Doing this by hand means writing a loop, checking each number, and adding the label one by one.

The Problem

This manual way is slow and boring. You have to write many lines of code, and it's easy to make mistakes like forgetting to check a number or mixing up labels. If the list is very long, it becomes a big headache.

The Solution

List comprehension with if-else lets you do this in one simple line. It checks each number and decides the label quickly and clearly. This saves time, reduces errors, and makes your code neat and easy to read.

Before vs After

✗ Before

result = []
for number in numbers:
    if number % 2 == 0:
        result.append('Even')
    else:
        result.append('Odd')

✓ After

result = ['Even' if number % 2 == 0 else 'Odd' for number in numbers]

What It Enables

This lets you transform lists with conditions in a clean, fast way, making your programs smarter and easier to write.

Real Life Example

Think about sorting emails into 'Important' or 'Not Important' based on keywords. List comprehension with if-else can quickly label each email, helping you organize your inbox automatically.

Key Takeaways

Manual loops for conditional list creation are long and error-prone.

List comprehension with if-else simplifies this into one clear line.

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

Practice

1. What does the following list comprehension do?
[x if x % 2 == 0 else -x for x in range(5)]

easy

A. Creates a list of numbers from 0 to 4 without changes

B. Creates a list of numbers where even numbers stay the same and odd numbers become negative

C. Creates a list of only negative numbers from 0 to 4

D. Creates a list of only even numbers from 0 to 4

Solution

1. Step 1: Understand the condition in the comprehension

   The condition x % 2 == 0 checks if a number is even.

2. Step 2: Apply the if-else for each number in range(5)

   If even, keep x; if odd, use -x. So 0,2,4 stay same; 1,3 become -1,-3.

3. Final Answer:

   Creates a list of numbers where even numbers stay the same and odd numbers become negative -> Option B

4. Quick Check:

   if-else picks value based on even check = D [OK]

Hint: If-else picks value before for loop [OK]

Common Mistakes:

• Thinking if-else goes after the for loop
• Confusing condition meaning (odd vs even)
• Ignoring the else part

2. Which of these is the correct syntax for a list comprehension with if-else in Python?

easy

A. [x if x > 0 else 0 for x in nums]

B. [x for x in nums if x > 0 else 0]

C. [if x > 0 then x else 0 for x in nums]

D. [x if x > 0 else 0 in nums]

Solution

1. Step 1: Recall correct if-else placement in list comprehension

   The if-else expression must come before the for loop inside the brackets.

2. Step 2: Check each option's syntax

   [x if x > 0 else 0 for x in nums] correctly places x if x > 0 else 0 before for x in nums. Others have syntax errors.

3. Final Answer:

   [x if x > 0 else 0 for x in nums] -> Option A

4. Quick Check:

   if-else before for loop = C [OK]

Hint: If-else must be before for in comprehension [OK]

Common Mistakes:

• Putting if-else after the for loop
• Using 'then' keyword (not in Python)
• Missing for loop entirely

3. What is the output of this code?

nums = [1, 2, 3, 4]
result = [x*2 if x % 2 == 0 else x+1 for x in nums]
print(result)

medium

A. [2, 4, 4, 6]

B. [2, 4, 4, 4]

C. [2, 4, 4, 8]

D. [2, 2, 4, 8]

Solution

1. Step 1: Evaluate each element with condition

   For each number: if even, multiply by 2; if odd, add 1. So 1->2, 2->4, 3->4, 4->8.

2. Step 2: Collect results in list

   The final list is [2, 4, 4, 6].

3. Final Answer:

   [2, 4, 4, 6] -> Option A

4. Quick Check:

   if even *2 else +1 = [2,4,4,6] [OK]

Hint: Check each item with condition carefully [OK]

Common Mistakes:

• Mixing up multiplication and addition
• Applying condition after the loop
• Wrong output list length

4. Find the error in this list comprehension:

values = [10, 15, 20]
new_values = [x if x > 15 else x*2 for x in values if x]

medium

A. The if condition after the for loop is invalid syntax

B. The if-else expression must be inside parentheses

C. No error, code runs fine

D. The else part should come after the for loop

Solution

1. Step 1: Analyze the list comprehension structure

   The if-else expression is before the for loop, and the trailing if filters values.

2. Step 2: Confirm syntax correctness

   Trailing if after for is allowed to filter items; no syntax error here.

3. Final Answer:

   No error, code runs fine -> Option C

4. Quick Check:

   Trailing if filters allowed = B [OK]

Hint: Trailing if filters items, allowed after for [OK]

Common Mistakes:

• Thinking trailing if is invalid
• Expecting else after for loop
• Misplacing parentheses

5. You have a list of temperatures in Celsius: temps = [22, -5, 0, 15, -10]. Use a list comprehension with if-else to create a new list where temperatures below 0 become the string 'Freezing', and others stay as numbers.
Which code does this correctly?

hard

A. [temp for temp in temps if temp >= 0 else 'Freezing']

B. [temp if temp < 0 else 'Freezing' for temp in temps]

C. [temp if temp > 0 else 'Freezing' for temp in temps]

D. ['Freezing' if temp < 0 else temp for temp in temps]

Solution

1. Step 1: Understand the condition and output

   Temperatures below 0 become 'Freezing', others stay numbers.

2. Step 2: Check each option's if-else placement and logic

   ['Freezing' if temp < 0 else temp for temp in temps] correctly uses 'Freezing' if temp < 0 else temp before the for loop. Others have wrong order or logic.

3. Final Answer:

   ['Freezing' if temp < 0 else temp for temp in temps] -> Option D

4. Quick Check:

   if-else before for, correct condition = A [OK]

Hint: Put if-else before for, condition matches output [OK]

Common Mistakes:

• Swapping if and else parts
• Placing if-else after for loop
• Using filter if instead of if-else expression