Basic list comprehension syntax in Python - Practice Problems & Coding Challenges

Challenge - 5 Problems

❓ Predict Output

intermediate

Output of simple list comprehension

What is the output of this Python code?

Python

result = [x * 2 for x in range(4)]
print(result)

A [0, 2, 4, 6]

B [1, 2, 3, 4]

C [2, 4, 6, 8]

D [0, 1, 2, 3]

💡 Hint

Think about what range(4) produces and how each element is multiplied by 2.

range(4) produces 0,1,2,3. Multiplying each by 2 gives 0,2,4,6.

❓ Predict Output

intermediate

List comprehension with condition

What is the output of this code?

Python

result = [x for x in range(6) if x % 2 == 0]
print(result)

A [1, 3, 5]

B [0, 1, 2, 3, 4, 5]

C [0, 2, 4]

D [2, 4, 6]

💡 Hint

The condition filters numbers divisible by 2.

The list comprehension includes only even numbers from 0 to 5, which are 0, 2, and 4.

❓ Predict Output

advanced

Nested list comprehension output

What is the output of this nested list comprehension?

Python

result = [(x, y) for x in range(2) for y in range(3)]
print(result)

A [(0, 0), (0, 1), (0, 2), (1, 0), (1, 1), (1, 2)]

B [(0, 0), (1, 0), (0, 1), (1, 1), (0, 2), (1, 2)]

C [(0, 0), (1, 0), (2, 0), (0, 1), (1, 1), (2, 1)]

D [(0, 0), (0, 1), (1, 0), (1, 1), (2, 0), (2, 1)]

💡 Hint

The first loop is x, the second is y, so y changes faster.

The comprehension pairs each x with every y in order, producing all combinations with x from 0 to 1 and y from 0 to 2.

❓ Predict Output

advanced

List comprehension with else clause (ternary)

What is the output of this code?

Python

result = [x if x % 2 == 0 else -x for x in range(5)]
print(result)

A [0, 1, 2, 3, 4]

B [0, -1, 2, -3, 4]

C [-0, 1, -2, 3, -4]

D [0, -1, -2, -3, -4]

💡 Hint

If x is even, keep it; if odd, make it negative.

For each number from 0 to 4, even numbers stay the same, odd numbers become negative.

🧠 Conceptual

expert

Understanding list comprehension order

Which option correctly describes the order in which elements are generated in this list comprehension?

result = [(x, y) for x in range(3) for y in range(2)]

A Only pairs where x equals y are included.

B For each y, all x values are paired before moving to the next y.

C Pairs are generated randomly without order.

D For each x, all y values are paired before moving to the next x.

💡 Hint

Think about how nested loops work in Python.

The comprehension is like nested loops: for each x, it loops through all y values, pairing them in order.

Practice

1. What does the following list comprehension do?
[x * 2 for x in range(3)]

easy

A. Creates a list of numbers from 1 to 6

B. Creates a list of numbers from 0 to 3

C. Creates a list of numbers doubled from 0 to 2

D. Creates a list of numbers squared from 0 to 2

Solution

1. Step 1: Understand the range and loop variable

   The range(3) generates numbers 0, 1, 2. The variable x takes these values one by one.

2. Step 2: Apply the expression to each value

   Each x is multiplied by 2, so the list becomes [0*2, 1*2, 2*2] = [0, 2, 4].

3. Final Answer:

   Creates a list of numbers doubled from 0 to 2 -> Option C

4. Quick Check:

   List comprehension doubles each number in range(3) [OK]

Hint: Remember: for x in range(3) gives 0,1,2 only [OK]

Common Mistakes:

• Thinking range(3) includes 3
• Confusing doubling with squaring
• Ignoring the for loop variable

2. Which of the following is the correct syntax for a list comprehension that creates a list of squares of numbers 0 to 4?

easy

A. [x**2 x in range(5)]

B. [for x in range(5) x**2]

C. [x^2 in range(5)]

D. [x**2 for x in range(5)]

Solution

1. Step 1: Recall list comprehension syntax

   The correct syntax is [expression for variable in iterable]. Here, expression is x**2 and iterable is range(5).

2. Step 2: Check each option

   [x**2 for x in range(5)] matches the correct syntax. Options A, B, and C have syntax errors or wrong order.

3. Final Answer:

   [x**2 for x in range(5)] -> Option D

4. Quick Check:

   Correct syntax is [expr for var in iterable] [OK]

Hint: Remember: list comprehension = [expression for variable in iterable] [OK]

Common Mistakes:

• Placing 'for' after expression
• Using ^ instead of ** for power
• Missing 'in' keyword

3. What is the output of this code?

nums = [1, 2, 3, 4]
squares = [n*n for n in nums if n % 2 == 0]
print(squares)

medium

A. [4, 16]

B. [1, 4, 9, 16]

C. [2, 4]

D. [1, 9]

Solution

1. Step 1: Understand the list and condition

   The list nums is [1, 2, 3, 4]. The condition 'if n % 2 == 0' selects even numbers: 2 and 4.

2. Step 2: Calculate squares of selected numbers

   Squares of 2 and 4 are 4 and 16, so the list squares is [4, 16].

3. Final Answer:

   [4, 16] -> Option A

4. Quick Check:

   Filter even numbers, then square them [OK]

Hint: Filter first, then apply expression in list comprehension [OK]

Common Mistakes:

• Including odd numbers by mistake
• Confusing values with their squares
• Ignoring the if condition

4. Find the error in this list comprehension:

result = [x + 1 for in range(5)]

medium

A. Missing variable name after 'for'

B. Using + instead of *

C. range(5) should be range(1,5)

D. List comprehension cannot use range

Solution

1. Step 1: Check the for loop syntax inside comprehension

   The syntax requires a variable name after 'for', like 'for x in range(5)'. Here, 'x' is missing.

2. Step 2: Identify the error

   Because the variable is missing, Python will raise a syntax error.

3. Final Answer:

   Missing variable name after 'for' -> Option A

4. Quick Check:

   for loop needs variable name [OK]

Hint: Always include variable after 'for' in comprehension [OK]

Common Mistakes:

• Omitting the loop variable
• Misplacing 'in' keyword
• Thinking range needs to start at 1

5. You have a list of words: words = ['apple', 'bee', 'cat', 'dog', 'elephant']. Using list comprehension, how do you create a list of the lengths of words that have more than 3 letters?

hard

A. [w for w in words if len(w) > 3]

B. [len(w) for w in words if len(w) > 3]

C. [len(words) for w in words if len(w) > 3]

D. [len(w) > 3 for w in words]

Solution

1. Step 1: Understand the filtering condition

   We want only words with length greater than 3, so use 'if len(w) > 3' to filter.

2. Step 2: Create list of lengths

   For each filtered word, get its length with len(w). So the comprehension is [len(w) for w in words if len(w) > 3].

3. Final Answer:

   [len(w) for w in words if len(w) > 3] -> Option B

4. Quick Check:

   Filter words by length, then get lengths [OK]

Hint: Filter first, then apply len() in comprehension [OK]

Common Mistakes:

• Returning words instead of lengths
• Using len(words) instead of len(w)
• Returning boolean instead of lengths