Scala 使用groupBy方法对列表进行分组

在本文中，我们将介绍Scala中的groupBy方法，它是一个非常有用且强大的方法，可以用于对列表中的元素进行分组。

阅读更多：Scala 教程

groupBy方法的基本用法

groupBy方法是Scala中List类的一个方法，它接收一个函数作为参数，并将列表中的元素按照该函数的返回值进行分组。返回值是一个Map，其中的key是分组的标准，value是被分组的元素列表。

下面是groupBy方法的基本语法：

def groupBy[K](f: (A) => K): Map[K, List[A]]

其中，f是一个函数，它接收一个列表元素作为参数，并返回该元素的分组标准的值。K是分组标准的类型。在分组结果中，K将作为Map的key，对应的元素列表将作为value。

让我们以一个实际的例子来说明groupBy方法的用法。假设我们有一个包含学生信息的列表，每个学生都有一个名称和年龄。我们希望根据学生的年龄对其进行分组。我们可以这样做：

val students = List(
  Student("Tom", 20),
  Student("Jerry", 18),
  Student("Alice", 20),
  Student("Bob", 19),
  Student("Cathy", 18)
)

case class Student(name: String, age: Int)

val groupedStudents = students.groupBy(_.age)

groupedStudents.foreach { case (age, students) =>
  println(s"Age:  $age, Students:$ {students.map(_.name).mkString(", ")}")
}

输出结果如下：

Age: 20, Students: Tom, Alice
Age: 18, Students: Jerry, Cathy
Age: 19, Students: Bob

通过上述例子，我们可以看出groupBy方法按照学生的年龄将学生分组，生成了一个Map。Map的key是学生的年龄，value是该年龄下的学生列表。我们遍历分组结果并打印出来，可以看到学生被正确地按照年龄进行了分组。

groupBy方法的高级用法

groupBy方法并不局限于只能按照一个属性进行分组，我们也可以按照多个属性进行分组。只需要将多个属性的返回值组合起来作为分组标准即可。

让我们以一个例子来说明groupBy方法的高级用法。假设我们有一个包含人员信息的列表，每个人都有一个名字、年龄和性别。我们希望按照性别和年龄对人员进行分组。我们可以这样做：

val persons = List(
  Person("Tom", 20, "Male"),
  Person("Jerry", 18, "Male"),
  Person("Alice", 20, "Female"),
  Person("Bob", 19, "Male"),
  Person("Cathy", 18, "Female")
)

case class Person(name: String, age: Int, gender: String)

val groupedPersons = persons.groupBy(person => (person.gender, person.age))

groupedPersons.foreach { case ((gender, age), persons) =>
  println(s"Gender:  $gender, Age:$ age, Persons: ${persons.map(_.name).mkString(", ")}")
}

输出结果如下：

Gender: Male, Age: 20, Persons: Tom
Gender: Male, Age: 18, Persons: Jerry
Gender: Female, Age: 20, Persons: Alice
Gender: Male, Age: 19, Persons: Bob
Gender: Female, Age: 18, Persons: Cathy

通过上述例子，我们可以看出groupBy方法按照人员的性别和年龄将人员分组，生成了一个Map。Map的key是一个元组，其中包含了性别和年龄两个属性的值，value是对应的人员列表。我们遍历分组结果并打印出来，可以看到人员被正确地按照性别和年龄进行了分组。