分组

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

你还可以调用带有第二个 lambda 实参的 groupBy() —— 即值转换函数。 在带有两个 lambda 的 groupBy() 结果 Map 中，由 keySelector 函数生成的键映射到值转换函数的结果，而不是原始元素。

此示例说明了如何使用 groupBy() 函数按首字母对字符串进行分组，使用 for 运算符遍历结果 Map 中的分组，然后使用 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 实例允许你以延迟方式对所有分组应用操作：这些分组实际上是在操作执行前才构建的。

具体而言，Grouping 支持以下操作：

• eachCount() 统计每个分组中的元素数量。
• fold() 和 reduce() 对每个分组作为单独的集合执行 fold and 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
    }
}