Scala 使用多个列进行分组 groupBy 的方法

在本文中，我们将介绍如何使用 Scala 集合中的多个列进行分组(groupBy)操作。

阅读更多：Scala 教程

什么是 groupBy？

groupBy 是 Scala 集合中非常常用的操作，它允许我们使用指定的条件将集合中的元素进行分组。通过对元素进行分组，我们可以获得每个分组的所有元素，并且在后续的操作中对分组进行处理。

使用单个列进行分组

首先，我们来看一下如何使用单个列进行分组。假设我们有一个包含学生姓名和年龄的列表，我们想要按照学生的年龄进行分组。下面是一个示例列表：

val students = List(("Alice", 18), ("Bob", 20), ("Alice", 19), ("Bob", 19), ("Charlie", 18))

现在我们使用年龄进行分组：

val groupedByAge = students.groupBy(_._2)

上述代码中，_._2 表示以元组的第二个元素（即年龄）进行分组。groupBy 方法将返回一个 Map，其中键(即年龄)对应的值是分组后的元素列表。我们来输出一下分组结果：

groupedByAge.foreach { case (age, students) =>
  println(s"Age: $age")
  students.foreach(println)
  println("----------------")
}

输出结果如下：

Age: 18
(Alice, 18)
(Charlie, 18)
----------------
Age: 20
(Bob, 20)
----------------
Age: 19
(Alice, 19)
(Bob, 19)
----------------

可以看到，我们已经成功地按照学生的年龄进行了分组。

使用多个列进行分组

现在，假设我们不仅仅希望按照学生的年龄进行分组，还希望按照姓名进行二级分组。换句话说，我们希望先按照年龄分组，然后在每个年龄组内再按照姓名分组。下面是我们的新列表：

val students = List(("Alice", 18), ("Bob", 20), ("Alice", 19), ("Bob", 19), ("Charlie", 18))

现在我们使用多个列进行分组：

val groupedByAgeAndName = students.groupBy { case (name, age) => (age, name) }

上述代码中，case (name, age) => (age, name) 表示我们按照年龄和姓名的顺序来生成分组键。注意，这里要使用元组来表示多个列。

与单个列分组类似，groupBy 方法返回一个 Map，其中键对应的值是分组后的元素列表。我们来输出分组结果：

groupedByAgeAndName.foreach { case ((age, name), students) =>
  println(s"Age:  $age, Name:$ name")
  students.foreach(println)
  println("----------------")
}

输出结果如下：

Age: 19, Name: Alice
(Alice, 19)
----------------
Age: 18, Name: Alice
(Alice, 18)
----------------
Age: 20, Name: Bob
(Bob, 20)
----------------
Age: 18, Name: Charlie
(Charlie, 18)
----------------
Age: 19, Name: Bob
(Bob, 19)
----------------

可以看到，我们已经成功地按照学生的年龄和姓名进行了分组。