[Swift] 고차 함수 CompactMap

기존의 어레이에서 값을 변형 시켜 새로운 어레이를 반환하는 것이 -> map

여기에 nil 값을 제거한 것이 compactMap 입니다

 

compactMap이 이제는 너무 몸에 익어버렸지만..

정리차원에서 실재로 어떻게 쓰일 수 있을지 정리했습니다!!

 

1. nil 제거하기

let test = [1, 2, nil, 4, 5, nil]

let removedNilTest = test.compactMap { $0 }

[1, 2, 4, 5]

CompactMap은 nil을 제외시키기 때문에 옵셔널을 사용해야 하는 환경에서 효과적으로 데이터를 핸들링 할 수 있어요! 만약 nil 값이 그대로 필요하다면 map을 쓰면 됩니다~

 

let sameWithAbove = test.compactMap { value in
    return value
}

고차 함수를 사용하게 되면 단축 변수를 사용할 수 있고 한 줄에 표현 가능해서 좀더 깔끔한 코드를 작성할 수 있어요. 하지만 코드가 복잡하고 직관성이 떨어지는 경우 이렇게 풀어서 적어주시는게 좋을 수도 있습니다!

 

2. 특정 값 추출

struct Car {
    let idx: Int
    let name: String?
}

let cars = [
    Car(idx: 0, name: "test1"),
    Car(idx: 1, name: "test2"),
    Car(idx: 2, name: nil)
]

print(cars.compactMap { $0.name })

["test1", "test2"]

구조체 어레이인 경우 구조체의 특정 값을 추출할 수 있어요!

idx만 필요하다면 $0.idx 로 리턴해주면 되겠죠?

 

3. 메소드 체이닝

enum OptionType {
    case optional
    case required
}

struct Option {
    let idx: String?
    let optionName: String?
    let price: String?
    let type: OptionType
}

let options = [
    Option(idx: "0", optionName: "빨강", price: "1000", type: .optional),
    Option(idx: "1", optionName: "검정", price: "2500", type: .required),
    Option(idx: "2", optionName: "파랑", price: "8000", type: .optional)
]

이 구조체를 계속 사용할 거에요!

 

(1) joined

let optionIdxs = options.compactMap { $0.idx }.joined(separator: ", ")
print(optionIdxs)

0, 1, 2

🧑‍💻 백엔드 개발자 : 선택된 옵션 값들의 idx를 쉼표로 구분해서 보내주세요~

이 상황에서 idx 만 추출하고 joined를 사용해서 하나의 String으로 만들어 줄 수 있어요

 

(2) sum

let optionPriceSum = options.compactMap { Int($0.price ?? "0") }.reduce(0, +)

11500

reduce도 고차 함수입니다!

reduce를 사용하면 어레이의 값의 합을 구할 수 있어요

reduce 앞까지가 [Int]이고 이 어레이의 각 요소의 합을 구하는 것이 

reduce(0, +)부분입니다

 

let sameWithAbove = options.compactMap { option in
    let price = option.price ?? "0"
    return Int(price)
}.reduce(0, +)

위 코드랑 같은 코드입니다~

 

4. 값 핸들링

let optionInfo = options.compactMap { "\($0.optionName ?? "(없음)"): \($0.price ?? "0")"}.joined(separator: "\n")

빨강: 1000 검정: 2500 파랑: 8000

값을 조합해서 리턴해 줄 수도 있어요

기존 어레이에서 새로운 값들을 가진 어레이로 변경가능합니다

 

let requireOptions = options.compactMap { $0.type == .required ? $0 : nil}

[Option(idx: Optional("1"), optionName: Optional("검정"), price: Optional("2500"), type: OptionType.required)]

특정한 값을 가지고 있는 값만 리턴해 줄 수 있어요!

만약 OptionType이 required 인 값만 어레이에 포함 시키고 싶다! 한다면

타입을 비교한 뒤 required 값을 가진 경우 그대로 리턴

아닌 경우 nil 을 리턴 하면

compactMap에 의해 nil이 제외되고 어레이가 생성됩니다

 

 

---

이번에 프로젝트 하면서 

nil로 절대 안오겠다는 값들이 nil로 와서 옵셔널 처리를 하다보니

map보다는 compactMap이 편리하더라구요

(

map은 O(N)이고 
compactMap은 O(M+N)

🥺🥲

)

 

어레이 속에 어레이 속에 어레이 안에 값

이런 식으로 데이터가 들어 있어서

고차 함수가 많이 도움이 됬던거 같아요..👍

 

 

문제나 궁금한거 있으시면 댓글 부탁드립니다!!!

 

https://www.pexels.com/ko-kr/photo/14968257/