Scala 如何在 groupBy 后将值聚合成集合

在本文中，我们将介绍Scala中如何使用groupBy方法对集合进行分组，并将分组后的值聚合成集合。

阅读更多：Scala 教程

groupBy 方法概述

在Scala中，groupBy方法是一个非常有用的集合方法。它接收一个函数作为参数，该函数用于对集合中的元素进行分组。groupBy方法返回一个Map，其中的键是分组的标准，值是具有相同标准的元素组成的集合。

groupBy方法的一般语法如下：

def groupBy[K](f: (A) ⇒ K): Map[K, Repr]

其中，f是一个用于生成键的函数，A是集合中的元素类型，K是用于分组的键的类型，Repr是生成的Map的值类型。

让我们看一个简单的示例来演示groupBy方法的用法。

val numbers = List(1, 2, 3, 4, 5, 6, 7, 8, 9, 10)
val groupedNumbers = numbers.groupBy(_ % 2)

在上面的示例中，我们使用groupBy方法将numbers列表中的元素按照它们对2取模的结果进行分组。这将返回一个Map，其中的键分别是0和1，对应的值是具有相同取模结果的数字组成的集合。

将值聚合为集合

一旦我们使用groupBy方法对集合进行分组，我们可以使用mapValues方法来集合分组后的值。

mapValues方法接收一个函数作为参数，该函数将应用于Map的值，并返回一个新的Map，其中的值是原始Map值经过转换后的结果。

让我们看一个示例来演示如何使用mapValues方法将分组后的值聚合为集合。

val fruits = List("apple", "banana", "orange", "pear", "blueberry", "kiwi")
val groupedFruits = fruits.groupBy(_.length)
val aggregatedFruits = groupedFruits.mapValues(_.toList)

在上面的示例中，我们对fruits列表中的元素按照它们的长度进行分组，并使用mapValues方法将每个分组的值转换为列表。

聚合后的结果将是一个Map，其中的键是水果的长度，对应的值是具有相同长度的水果组成的列表。

自定义聚合逻辑

除了使用mapValues方法将分组后的值转换为集合，我们还可以使用其他自定义的聚合逻辑。

例如，如果我们想对分组后的值进行求和，可以使用foldLeft方法来实现。

val numbers = List(1, 2, 3, 4, 5, 6, 7, 8, 9, 10)
val groupedNumbers = numbers.groupBy(_ % 2)
val sumOfGroupedNumbers = groupedNumbers.mapValues(_.foldLeft(0)(_ + _))

在上面的示例中，我们首先对numbers列表中的元素按照它们对2取模的结果进行分组。然后，我们使用mapValues方法将分组后的值转换为它们的总和。

最终的结果将是一个Map，其中的键是0和1，对应的值分别是偶数和奇数的总和。