[ Swift ] 고차함수 map, compactMap, flatMap의 차이

 

 

오늘은 고차함수 map과 compactMap, flatMap에 대해 알아봅시다!

 

 

고차함수는 함수를 매개변수로 사용하거나 리턴값으로 함수를 출력하는 함수입니다.

클로저를 사용한 함수라는 뜻이겠죠?!?

 

 

우선 map이 어떤 함수인지에 대한 이해부터 해봅시다!

map

map 은 함수에서 반환된 값들을 배열로 리턴해주는 함수입니다.

 

 

예를 들어 1부터 10까지의 수를 제곱한 값을 배열에 저장한다고 해봅시다.

고차함수를 쓰지 않는다면,

 

var array = [Int]()

for i in 1...10 {
    array.append(i*i)
}

print(array)
// 출력: [1, 4, 9, 16, 25, 36, 49, 64, 81, 100]

 

이렇게 구현해야겠죠?!

이를 한 문장으로 줄여줄 수 있는 친구가 이 map이라는 고차함수입니다.

 

let arrayMap = (1...10).map{ $0 * $0 }

print(arrayMap)
// 출력: [1, 4, 9, 16, 25, 36, 49, 64, 81, 100]

map의 클로저에서 반환된 친구들을 알아서 배열로 선언하여 출력해 줍니다. 

 

for문보다 map의 실행 시간이 더 빠르고 코드를 보다 간결하게 작성할 수 있어서 많이 사용이 돼요!

 

 

 [ 예시 1 ] 

let alphabet = "abcdefghijklmnopqrstuvwxyz"
print(alphabet.map{ $0 })
// 출력: ["a", "b", "c", "d", "e", "f", "g", "h", "i", "j", "k", "l", "m", "n", "o", "p", "q", "r", "s", "t", "u", "v", "w", "x", "y", "z"]

 

[ 예시 2 ]

var word = ["apple", "banana", "cinamon", "delmont", "element"]
print(word.enumerated().map{ String($0.offset)+$0.element })
// 출력: ["0apple", "1banana", "2cinamon", "3delmont", "4element"]

 

[ 예시 3 ]

var students = [ 1: "james", 2: "tim", 3:"john", 4:"brian"]
print(students.map{ $0.value })
// 출력: ["john", "tim", "brian", "james"]

 

 

compactMap

compactMap 은 map과 동일한 기능을 하는 함수이지만 nil 을 제외한 출력값을 배열로 선언해 출력해요. 

예를 들어, 문자열 배열을 Int로 바꾸어서 출력하려고 할 경우

 

let isItNumArray = ["one", "2", "three", "///4///", "5!"]

print(isItNumArray.map{ Int($0) })
// 출력: [nil, Optional(2), nil, nil, nil]

print(isItNumArray.compactMap{ Int($0) })
// 출력: [2]

 

map 과 compactMap 의 출력이 다르다는 것을 볼 수 있습니다. map은 nil을 출력하고 nil이 아니라면

unwrapping 되지

않은 옵셔널 값을 출력해요. 그에 반해 compactMap 은 nil값을 제거하며 옵셔널 바인딩을 하여 출력합니다. 그래서 compactMap을 사용하면 옵셔널값도 foced unwrapping 을 하지 않고 unwrapping 할 수 있어서 더 안전하게 값을 출력가능한거죠.

 

 

flatMap

flatMap 은 함수에서 반환된 값들을 배열로 리턴해주지만 map과 다른 점은 flatten하게 만들어주고, compactMap 과 동일하게 nil값을 제거하며 옵셔널 바인딩을 해줍니다.

여기서 flatten 이라는 것은 2차원 배열에서 1차원 배열로 평평하게 만들어준다는 것인데요. 위의 예시를 보면 이해가 갈 것입니다. 

 

print( (0...5).map { Array(repeating: $0, count: $0) } )
// 출력: [[], [1], [2, 2], [3, 3, 3], [4, 4, 4, 4], [5, 5, 5, 5, 5]]

print( (0...5).flatMap { Array(repeating: $0, count: $0) } )
// 출력: [1, 2, 2, 3, 3, 3, 4, 4, 4, 4, 5, 5, 5, 5, 5]

 

Q : flatMap 과 compactMap 은 둘 다 옵셔널 바인딩을 해주면서 nil 값을 제거하는 역할을 하는데 그럼 언제 flatMap 을 사용하고 언제 compactMap 을 사용할까요?

 

A : 2차원 배열에서 1차원 배열로 flatten 하게 만들고 싶다면 flatMap 을 사용하고, 2차원 배열을 유지하고 싶다면 compactMap 을 사용하면 돼요!

 

 

 

이렇게 오늘 포스팅을 마치도록 하겠습니다😊

만약 부족한 점이 있다면 댓글로 피드백해 주시면 감사하겠습니다🩵

 

 

 

 

📄 참고 블로그 1 https://withthemilkyway.tistory.com/61
📄 참고 블로그 2 https://levenshtein.tistory.com/502#google_vignette