CompactMap for optional unwrapping in Swift - Step-by-Step Execution

Concept Flow - CompactMap for optional unwrapping

Start with array of optionals

↓

Apply compactMap

↓

Check each element

↓

Not nil

↓

Keep value

↓

Return array without nils

compactMap takes an array with optional values, removes all nils, and returns an array of only the unwrapped values.

Execution Sample

Swift

let numbers: [Int?] = [1, nil, 3, nil, 5]
let unwrapped = numbers.compactMap { $0 }
print(unwrapped)

This code removes nil values from the array and prints only the non-nil numbers.

Execution Table

Step	Element	Is nil?	Action	Resulting Array
1	1	No	Keep 1	[1]
2	nil	Yes	Discard nil	[1]
3	3	No	Keep 3	[1, 3]
4	nil	Yes	Discard nil	[1, 3]
5	5	No	Keep 5	[1, 3, 5]
End	-	-	All elements processed	[1, 3, 5]

💡 All elements checked; nil values removed; final array contains only unwrapped Ints.

Variable Tracker

Variable	Start	After 1	After 2	After 3	After 4	After 5	Final
numbers	[1, nil, 3, nil, 5]	[1, nil, 3, nil, 5]	[1, nil, 3, nil, 5]	[1, nil, 3, nil, 5]	[1, nil, 3, nil, 5]	[1, nil, 3, nil, 5]	[1, nil, 3, nil, 5]
unwrapped	[]	[1]	[1]	[1, 3]	[1, 3]	[1, 3, 5]	[1, 3, 5]

Key Moments - 2 Insights

Why does compactMap remove nil values but map does not?

compactMap unwraps optionals and discards nils, while map keeps nils as they are. See execution_table rows 2 and 4 where nils are discarded.

What happens if all elements are nil?

compactMap returns an empty array because all nils are discarded. This is shown by the exit_note where only non-nil values remain.

Visual Quiz - 3 Questions

Look at the execution_table, what is the resulting array after step 3?

A [1, nil, 3]

B [1]

C [1, 3]

D [1, 3, nil]

💡 Hint

Check the 'Resulting Array' column at step 3 in the execution_table.

At which step does compactMap discard the first nil value?

A Step 2

B Step 1

C Step 3

D Step 4

💡 Hint

Look at the 'Is nil?' and 'Action' columns in the execution_table.

If the input array had no nils, how would the final unwrapped array compare to the original?

A It would be empty

B It would be the same as the original array without optionals

C It would contain only nils

D It would contain fewer elements

💡 Hint

Refer to variable_tracker and concept_flow about how compactMap works with non-nil values.

Concept Snapshot

compactMap removes nils from an array of optionals.
Syntax: array.compactMap { $0 }
Returns an array of unwrapped non-nil values.
Useful to clean arrays with optional elements.
map keeps nils; compactMap removes them.

Full Transcript

We start with an array of optional integers, some are nil. Using compactMap, we check each element. If it is not nil, we keep it; if it is nil, we discard it. Step by step, the resulting array grows only with non-nil values. At the end, we have an array without any nils. This is different from map, which keeps nils. compactMap is useful to unwrap optionals and clean arrays in one step.

Practice

1. What does the compactMap function do when used on an array of optionals in Swift?

easy

A. It unwraps optionals and removes all nil values from the array.

B. It converts all values to strings regardless of their type.

C. It sorts the array in ascending order.

D. It replaces all nil values with zero.

Solution

1. Step 1: Understand what compactMap does

   compactMap takes an array of optionals and unwraps each element, removing any nil values.

2. Step 2: Compare with other options

   It does not convert types, sort, or replace nil with zero. It only filters out nil and unwraps values.

3. Final Answer:

   It unwraps optionals and removes all nil values from the array. -> Option A

4. Quick Check:

   compactMap removes nil = B [OK]

Hint: Remember: compactMap unwraps and drops nils [OK]

Common Mistakes:

• Confusing compactMap with map (which keeps nils)
• Thinking compactMap sorts or changes types
• Assuming compactMap replaces nil with default values

2. Which of the following is the correct syntax to use compactMap on an array numbers of type [Int?] to get an array of non-optional Int values?

easy

A. let result = numbers.compactMap { $0 }

B. let result = numbers.compactMap { Int($0) }

C. let result = numbers.compactMap { $0! }

D. let result = numbers.compactMap { String($0) }

Solution

1. Step 1: Understand the input and output types

   numbers is an array of optional integers [Int?]. We want to unwrap and remove nil values to get [Int].

2. Step 2: Check each option's correctness

   let result = numbers.compactMap { $0 } uses compactMap { $0 } which unwraps optionals and removes nil. let result = numbers.compactMap { Int($0) } tries to convert optional Int to Int again, which is unnecessary. let result = numbers.compactMap { $0! } force unwraps, which can crash if nil. let result = numbers.compactMap { String($0) } converts to String, changing type.

3. Final Answer:

   let result = numbers.compactMap { $0 } -> Option A

4. Quick Check:

   Use compactMap with identity closure to unwrap optionals [OK]

Hint: Use compactMap { $0 } to unwrap optionals safely [OK]

Common Mistakes:

• Using force unwrap inside compactMap causing runtime errors
• Trying to convert optionals unnecessarily
• Changing the type unintentionally

3. What is the output of the following Swift code?

let values: [Int?] = [1, nil, 3, nil, 5]
let unwrapped = values.compactMap { $0 }
print(unwrapped)

medium

A. Compilation error

B. [1, 3, 5]

C. [1, nil, 3, nil, 5]

D. [Optional(1), Optional(3), Optional(5)]

Solution

1. Step 1: Understand the input array

   The array values contains optional integers with some nil values.

2. Step 2: Apply compactMap

   compactMap { $0 } unwraps optionals and removes nil values, resulting in [1, 3, 5].

3. Final Answer:

   [1, 3, 5] -> Option B

4. Quick Check:

   compactMap removes nils and unwraps = [1, 3, 5] [OK]

Hint: compactMap removes nils and unwraps optionals [OK]

Common Mistakes:

• Expecting optionals in output instead of unwrapped values
• Thinking nils remain in the output
• Confusing compactMap with map

4. Identify the error in this Swift code snippet and choose the correct fix:

let strings: [String?] = ["apple", nil, "banana"]
let lengths = strings.compactMap { $0.count }
print(lengths)

medium

A. Error: Use force unwrap $0! before accessing count

B. Error: compactMap should be replaced with map

C. No error; code runs correctly

D. Error: Cannot use $0.count on optional; fix by using compactMap { $0?.count }

Solution

1. Step 1: Analyze the closure inside compactMap

   The closure uses $0.count, but $0 is an optional String?. Accessing count directly on optional causes error.

2. Step 2: Correct the closure to safely unwrap

   Using $0?.count returns an optional Int, which compactMap unwraps and removes nils, fixing the error.

3. Final Answer:

   Error: Cannot use $0.count on optional; fix by using compactMap { $0?.count } -> Option D

4. Quick Check:

   Use optional chaining inside compactMap closure [OK]

Hint: Use optional chaining ($0?.property) inside compactMap [OK]

Common Mistakes:

• Trying to access property directly on optional causing error
• Using map instead of compactMap losing nil filtering
• Force unwrapping causing runtime crashes

5. Given a dictionary [String: Int?] representing item stock counts, how can you create a new dictionary with only items that have a non-nil stock count using compactMap or similar methods?

hard

A. Use filter { $0.value != nil } without unwrapping values.

B. Use mapValues { $0! } to force unwrap all values.

C. Use compactMapValues { $0 } to unwrap and filter out nil values.

D. Use compactMap { $0 } directly on the dictionary.

Solution

1. Step 1: Understand dictionary optional values

   The dictionary has optional Int values. We want to remove keys with nil values and unwrap the rest.

2. Step 2: Use compactMapValues to unwrap and filter

   compactMapValues { $0 } unwraps optionals in values and removes entries with nil values, producing a dictionary with non-optional Int values.

3. Final Answer:

   Use compactMapValues { $0 } to unwrap and filter out nil values. -> Option C

4. Quick Check:

   compactMapValues unwraps and filters dictionary values [OK]

Hint: Use compactMapValues to unwrap and remove nil dictionary values [OK]

Common Mistakes:

• Using mapValues with force unwrap risking crashes
• Filtering without unwrapping values
• Trying to use compactMap directly on dictionary