Android Kotlin 中的 `groupBy` 方法详解

在 Kotlin 中，groupBy 是一个非常有用的集合操作函数。我们可以使用它按照某个标准，将集合中的元素分组，形成一个 Map，其中 key 是我们分组的标准，value 是符合这个标准的元素列表。本文将通过几个实际例子，来详细说明如何使用 groupBy，并且还会解释一些常见的应用场景。

什么是 groupBy？

groupBy 是 Kotlin 标准库中的一个扩展函数，适用于集合、列表、数组等。它接收一个 lambda 表达式，根据这个 lambda 表达式的返回值对集合中的元素进行分组。

语法

fun <T, K> Iterable<T>.groupBy(
keySelector: (T) -> K
): Map<K, List<T>>

• T 表示集合中的元素类型
• K 表示分组的标准，也就是我们通过 keySelector 函数返回的值
• 返回值是一个 Map<K, List<T>>，其中 K 是分组标准，List<T> 是符合该标准的元素列表

😄 简单来说，groupBy 就是将集合中的元素按照我们提供的标准进行分组。

基本用法

假设我们有一个员工列表，其中包含每个员工的姓名和年龄。我们想要按照员工的年龄来分组，那么可以这样使用 groupBy：

data class Employee(val name: String, val age: Int)

fun main() {
 val employees = listOf(
     Employee("John", 25),
     Employee("Sarah", 30),
     Employee("Mike", 25),
     Employee("Emma", 30),
     Employee("Chris", 22)
 )

 val groupedByAge = employees.groupBy { it.age }

 println(groupedByAge)
}

输出结果

{25=[Employee(name=John, age=25), Employee(name=Mike, age=25)], 
30=[Employee(name=Sarah, age=30), Employee(name=Emma, age=30)], 
22=[Employee(name=Chris, age=22)]}

在这个例子中，我们根据员工的 age 属性进行分组，得到的结果是一个 Map，每个年龄对应一个员工列表。

自定义分组标准

如果我们想根据员工的名字长度来分组，可以通过 groupBy 函数轻松实现：

val groupedByNameLength = employees.groupBy { it.name.length }

println(groupedByNameLength)

输出结果

{4=[Employee(name=John, age=25), Employee(name=Mike, age=25)], 
5=[Employee(name=Sarah, age=30), Employee(name=Chris, age=22)], 
4=[Employee(name=Emma, age=30)]}

通过这个例子，我们将员工按名字长度进行了分组。

常见应用场景

1. 按条件统计

比如我们有一个产品列表，想按照产品的类型分组统计每种类型的产品数量：

data class Product(val name: String, val type: String)

val products = listOf(
 Product("Laptop", "Electronics"),
 Product("Phone", "Electronics"),
 Product("Sofa", "Furniture"),
 Product("Table", "Furniture"),
 Product("Headphones", "Electronics")
)

val groupedByType = products.groupBy { it.type }

val countByType = groupedByType.mapValues { it.value.size }

println(countByType)

输出结果

{Electronics=3, Furniture=2}

通过这个方式，我们可以轻松统计出每种类型的产品数量。

2. 按属性分类

在某些场景下，我们需要将数据按属性分组，比如按照学生的成绩等级分组：

data class Student(val name: String, val score: Int)

val students = listOf(
 Student("Alice", 85),
 Student("Bob", 70),
 Student("Charlie", 90),
 Student("David", 60)
)

val groupedByGrade = students.groupBy { 
 when {
     it.score >= 85 -> "A"
     it.score >= 70 -> "B"
     else -> "C"
 }
}

println(groupedByGrade)

输出结果

{A=[Student(name=Alice, score=85), Student(name=Charlie, score=90)], 
B=[Student(name=Bob, score=70)], 
C=[Student(name=David, score=60)]}

通过 when 表达式，我们可以根据学生的分数将他们分为 A、B、C 三个等级。

结论

Kotlin 中的 groupBy 是一个功能强大的工具，可以让我们轻松地按照不同标准对集合中的元素进行分组。无论是按属性分组还是按自定义规则分组，groupBy 都能够提供灵活的解决方案。在实际开发中，它能够帮助我们更高效地处理和组织数据，让代码更加简洁易读。

Best Regards!!