그룹화

코틀린 표준 라이브러리는 컬렉션 요소를 그룹화하기 위한 확장 함수를 제공합니다. 기본 함수인 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")
    }
    // 키: O, 값: [one]
    // 키: T, 값: [two, three]
    // 키: F, 값: [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 인스턴스는 모든 그룹에 연산을 지연 방식으로 적용할 수 있게 합니다. 즉, 그룹은 연산이 실행되기 직전에 실제로 빌드됩니다.

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

• eachCount()는 각 그룹의 요소를 계산합니다.
• fold()와 reduce()는 각 그룹에 대해 별도의 컬렉션으로 폴드 및 리듀스 연산을 수행하고 결과를 반환합니다.
• aggregate()는 각 그룹의 모든 요소에 주어진 연산을 순차적으로 적용하고 결과를 반환합니다. 이는 Grouping에 대해 어떤 연산이든 수행하는 일반적인 방법입니다. 폴드 또는 리듀스로 충분하지 않을 때 사용자 지정 연산을 구현하는 데 사용하세요.

결과 Map에 for 연산자를 사용하여 groupingBy() 함수에 의해 생성된 그룹들을 반복할 수 있습니다. 이를 통해 각 키와 해당 키와 관련된 요소의 개수에 접근할 수 있습니다.

다음 예시는 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")
        // 키: o, 개수: 1
        // 키: t, 개수: 2
        // 키: f, 개수: 2
    }
}