본문으로 건너뛰기

그룹화

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

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

이 예제는 groupBy() 함수를 사용하여 문자열을 첫 글자로 그룹화하고, 결과 Map에서 for 연산자를 사용하여 그룹을 반복한 다음, keySelector 함수를 사용하여 값을 대문자로 변환하는 방법을 보여줍니다.

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 인스턴스를 사용하면 모든 그룹에 대해 지연 방식으로 작업을 적용할 수 있습니다. 그룹은 작업 실행 직전에 실제로 빌드됩니다.

특히 Grouping은 다음 작업을 지원합니다.

  • eachCount()는 각 그룹의 요소 수를 계산합니다.
  • fold()reduce()는 각 그룹에 대해 별도의 컬렉션으로 fold 및 reduce 작업을 수행하고 결과를 반환합니다.
  • aggregate()는 주어진 작업을 각 그룹의 모든 요소에 연속적으로 적용하고 결과를 반환합니다. 이것은 Grouping에 대한 모든 작업을 수행하는 일반적인 방법입니다. fold 또는 reduce가 충분하지 않은 경우 사용자 정의 작업을 구현하는 데 사용합니다.

결과 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")
// Key: o, Count: 1
// Key: t, Count: 2
// Key: f, Count: 2
}

}