Skip to content

グループ化

Kotlin標準ライブラリは、コレクション要素をグループ化するための拡張関数を提供します。 基本的な関数groupBy()はラムダ関数を受け取り、Mapを返します。 このマップでは、各キーはラムダの結果であり、対応する値はその結果が返される要素のListです。 この関数は、例えばStringのリストをその最初の文字でグループ化するために使用できます。

groupBy()は、2番目のラムダ引数(値変換関数)とともに呼び出すこともできます。 2つのラムダを持つgroupBy()の結果マップでは、keySelector関数によって生成されたキーは、元の要素ではなく値変換関数の結果にマップされます。

この例では、groupBy()関数を使用して文字列を最初の文字でグループ化し、結果のMap上のグループをfor演算子で反復処理し、valueTransform関数を使用して値を大文字に変換する方法を示します。

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

    // Groups the strings by their first letter using groupBy()
    val groupedByFirstLetter = numbers.groupBy { it.first().uppercase() }
    println(groupedByFirstLetter)
    // {O=[one], T=[two, three], F=[four, five]}

    // Iterates through each group and prints the key and its associated values
    for ((key, value) in groupedByFirstLetter) {
        println("Key: $key, Values: $value")
    }
    // Key: O, Values: [one]
    // Key: T, Values: [two, three]
    // Key: F, Values: [four, five]

    // Groups the strings by their first letter and transforms the values to uppercase
    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で十分でない場合に、カスタム操作を実装するために使用します。

groupingBy()関数によって作成されたグループを反復処理するには、結果のMap上でfor演算子を使用できます。 これにより、各キーと、そのキーに関連付けられた要素のカウントにアクセスできます。

次の例は、groupingBy()関数を使用して文字列を最初の文字でグループ化し、各グループの要素をカウントし、その後各グループを反復処理してキーと要素のカウントを出力する方法を示しています。

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

    // Groups the strings by their first letter using groupingBy() and counts the elements in each group
    val grouped = numbers.groupingBy { it.first() }.eachCount()

    // Iterates through each group and prints the key and its associated values
    for ((key, count) in grouped) {
        println("Key: $key, Count: $count")
        // Key: o, Count: 1
        // Key: t, Count: 2
        // Key: f, Count: 2
    }
}