Kotlin集合过滤-优快云博客

过滤操作

在处理集合时，根据特定条件过滤集合或检查集合中是否包含符合特定条件的元素是软件开发中的常见任务。为了解决这个问题，我们可以使用 Kotlin 中实现的函数式 API。

在本主题中，我们将介绍如何使用谓词过滤集合，并获得满足给定谓词的新集合。

过滤

Kotlin 提供了一个函数式 API，它使用扩展函数，允许我们根据谓词过滤集合中的元素。filter() 函数接受一个谓词，并返回一个新的集合，其中包含满足谓词（或条件）的元素。此外，我们还可以使用 filterIndexed()，该函数接受元素的索引和元素本身，并返回一个符合谓词的集合。还有 filterNot()，它返回一个不满足谓词的集合。这些方法适用于所有基本的 Kotlin 集合（列表、集合和映射）。

val isEven: (Int) -> Boolean = { x -> x % 2 == 0 }

fun main() {
    val numbers = listOf(0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10)

    // 获取偶数列表
    println(numbers.filter { x -> x % 2 == 0 }) // [0, 2, 4, 6, 8, 10]
    println(numbers.filter { isEven(it) }) // [0, 2, 4, 6, 8, 10]
    println(numbers.filter(isEven)) // [0, 2, 4, 6, 8, 10]

    // 获取奇数列表
    println(numbers.filter { x -> x % 2 != 0 }) // [1, 3, 5, 7, 9]
    
    // 使用 filterNot 获取奇数列表
    println(numbers.filterNot { x -> x % 2 == 0 }) // [1, 3, 5, 7, 9]
   
    // 获取索引大于 3 且为偶数的数字
    println(numbers.filterIndexed { i, number -> i > 3 && number % 2 == 0 }) // [4, 6, 8, 10]

    // 获取索引为奇数且以 "k" 开头的单词
    val words = listOf("peter", "kyle", "robert", "kate", "kevin", "anne", "jeremy")
    println(words.filterIndexed { i, word -> i % 2 != 0 && word.startsWith("k") }) // [kyle, kate]
}

解释代码

函数式编程中的 filter 函数是一个有效的工具，它可以代替命令式编程中的 for 和 if：

fun main() {
    val numbers = listOf(0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10)
    
    val even = numbers.filter { x -> x % 2 == 0 }

    val odd = mutableListOf<Int>()
    for (i in numbers) {
        if (i % 2 != 0) {
            odd.add(i)
        }
    }

    println(even) // [0, 2, 4, 6, 8, 10]
    println(odd) // [1, 3, 5, 7, 9]
}

过滤和类型

我们还可以使用一些函数，通过元素的类型来过滤集合。filterIsInstance() 函数根据给定的类型谓词返回一个新的集合，filterNotNull() 函数则返回所有非 null 元素。

filterIsInstance 函数不接受一个显式的谓词函数，而是通过类型参数来指定过滤的类型。

fun main() {
    val elements = listOf(null, 0, "string", 3.14, null, 'c', "Luke")

    // 只获取字符串元素
    println(elements.filter { x -> x is String }) // [string, Luke]
    println(elements.filterIsInstance<String>()) // [string, Luke]
    // 获取非 null 元素
    println(elements.filterNot { x -> x == null }) // [0, string, 3.14, c, Luke]
    println(elements.filterNotNull()) // [0, string, 3.14, c, Luke]
    // 获取整型元素
    println(elements.filterIsInstance<Int>()) // [0]
}

分区

partition() 函数将原始集合分为两个部分：一个包含符合谓词的元素，另一个包含不符合谓词的元素。

fun main() {
    val numbers = listOf(0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10)

    val (even, odd) = numbers.partition { x -> x % 2 == 0 }
    println(even) // [0, 2, 4, 6, 8, 10]
    println(odd) // [1, 3, 5, 7, 9]
}