그룹화

Kotlin 표준 라이브러리는 컬렉션 요소를 그룹화하기 위한 확장 함수를 제공합니다. 기본 함수인 groupBy()는 람다 함수를 인자로 받아 Map을 반환합니다. 이 맵에서 각 키는 람다의 결과이며, 해당 값은 이 결과가 반환된 요소들의 List입니다. 예를 들어, 이 함수를 사용해 String 리스트를 첫 글자에 따라 그룹화할 수 있습니다.

또한, 두 번째 람다 인자인 값 변환 함수와 함께 groupBy()를 호출할 수 있습니다. 두 개의 람다를 사용하는 groupBy()의 결과 맵에서 keySelector 함수로 생성된 키는 원본 요소 대신 값 변환 함수의 결과에 매핑됩니다.

이 예제는 groupBy() 함수를 사용하여 문자열을 첫 글자로 그룹화하고, 결과 Map의 그룹들을 for 연산자로 순회하며, valueTransform 함수를 사용하여 값을 대문자로 변환하는 방법을 보여줍니다:

fun main() {
    val numbers = listOf("one", "two", "three", "four", "five")

    // groupBy()를 사용하여 문자열을 첫 글자로 그룹화합니다.
    val groupedByFirstLetter = numbers.groupBy { it.first().uppercase() }
    println(groupedByFirstLetter)
    // {O=[one], T=[two, three], F=[four, five]}

    // 각 그룹을 순회하며 키와 연결된 값들을 출력합니다.
    for ((key, value) in groupedByFirstLetter) {
        println("Key: $key, Values: $value")
    }
    // Key: O, Values: [one]
    // Key: T, Values: [two, three]
    // Key: F, Values: [four, five]

    // 문자열을 첫 글자로 그룹화하고 값을 대문자로 변환합니다.
    val groupedAndTransformed = numbers.groupBy(keySelector = { it.first() }, valueTransform = { it.uppercase() })
    println(groupedAndTransformed)
    // {o=[ONE], t=[TWO, THREE], f=[FOUR, FIVE]}
}

요소를 그룹화한 다음 모든 그룹에 한 번에 연산을 적용하려면 groupingBy() 함수를 사용하세요. 이 함수는 Grouping 타입의 인스턴스를 반환합니다. Grouping 인스턴스를 사용하면 지연(lazy) 방식으로 모든 그룹에 연산을 적용할 수 있습니다. 즉, 그룹은 연산이 실행되기 직전에 실제로 생성됩니다.

구체적으로 Grouping은 다음 연산들을 지원합니다:

• eachCount()는 각 그룹의 요소 개수를 셉니다.
• fold() 및 reduce()는 각 그룹을 별개의 컬렉션으로 보고 각 그룹에 대해 fold 및 reduce 연산을 수행하고 그 결과를 반환합니다.
• aggregate()는 각 그룹의 모든 요소에 주어진 연산을 순차적으로 적용하고 그 결과를 반환합니다. 이는 Grouping에서 임의의 연산을 수행하는 일반적인 방법입니다. fold나 reduce만으로 충분하지 않을 때 커스텀 연산을 구현하기 위해 사용하세요.

groupingBy() 함수로 생성된 그룹들을 순회하기 위해 결과 Map에서 for 연산자를 사용할 수 있습니다. 이를 통해 각 키와 해당 키에 연결된 요소의 개수에 접근할 수 있습니다.

다음 예제는 groupingBy() 함수를 사용하여 문자열을 첫 글자로 그룹화하고, 각 그룹의 요소 개수를 센 다음, 각 그룹을 순회하며 키와 요소 개수를 출력하는 방법을 보여줍니다:

fun main() {
    val numbers = listOf("one", "two", "three", "four", "five")

    // groupingBy()를 사용하여 문자열을 첫 글자로 그룹화하고 각 그룹의 요소 개수를 셉니다.
    val grouped = numbers.groupingBy { it.first() }.eachCount()

    // 각 그룹을 순회하며 키와 연결된 값들을 출력합니다.
    for ((key, count) in grouped) {
        println("Key: $key, Count: $count")
        // Key: o, Count: 1
        // Key: t, Count: 2
        // Key: f, Count: 2
    }
}