分组

Kotlin 标准库提供了用于对集合元素进行分组的扩展函数。基本函数 groupBy() 接收一个 lambda 函数并返回一个 Map。在这个 Map 中，每个键都是 lambda 的结果，对应的值是返回该结果的元素 List。例如，此函数可用于按 String 的首字母对 String 列表进行分组。

你也可以传入第二个 lambda 参数（一个值转换函数）来调用 groupBy()。在使用两个 lambda 的 groupBy() 的结果 Map 中，keySelector 函数生成的键将映射到值转换函数的结果，而不是原始元素。

此示例演示了如何使用 groupBy() 函数按字符串的首字母对其进行分组，使用 for 操作符遍历结果 Map 中的组，然后使用 valueTransform 函数将值转换为大写：

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() 对每个组作为单独的集合执行折叠和归约操作并返回结果。
• aggregate() 依次将给定操作应用于每个组中的所有元素并返回结果。 这是对 Grouping 执行任何操作的通用方法。当折叠或归约不足以满足需求时，可以使用它来实现自定义操作。

你可以在结果 Map 上使用 for 操作符来遍历由 groupingBy() 函数创建的组。这允许你访问每个键以及与该键关联的元素计数。

以下示例演示了如何使用 groupingBy() 函数按字符串的首字母对其进行分组，计算每个组中的元素，然后遍历每个组以打印键和元素计数：

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
    }
}