过滤集合

过滤是集合处理中最常见的任务之一。在 Kotlin 中，过滤条件由谓词（predicates）定义 – 谓词是接受集合元素并返回布尔值的 lambda 函数：true 表示给定的元素匹配谓词，false 表示相反。

标准库包含一组扩展函数，允许你通过一次调用来过滤集合。这些函数不会更改原始集合，因此它们可用于可变和只读集合。要操作过滤结果，你应该将其赋值给变量或在过滤后链式调用函数。

按谓词过滤

基本的过滤函数是 filter()。当使用谓词调用时，filter() 返回与谓词匹配的集合元素。对于 List 和 Set，结果集合都是 List，对于 Map，结果也是 Map。

fun main() {

    val numbers = listOf("one", "two", "three", "four")  
    val longerThan3 = numbers.filter { it.length > 3 }
    println(longerThan3)

    val numbersMap = mapOf("key1" to 1, "key2" to 2, "key3" to 3, "key11" to 11)
    val filteredMap = numbersMap.filter { (key, value) `->` key.endsWith("1") && value > 10}
    println(filteredMap)

}

filter() 中的谓词只能检查元素的值。如果要在过滤器中使用元素位置，请使用 filterIndexed()。它接受一个带有两个参数的谓词：元素的索引和值。

要按否定条件过滤集合，请使用 filterNot()。它返回谓词产生 false 的元素的列表。

fun main() {

    val numbers = listOf("one", "two", "three", "four")
    
    val filteredIdx = numbers.filterIndexed { index, s `->` (index != 0) && (s.length < 5)  }
    val filteredNot = numbers.filterNot { it.length <= 3 }

    println(filteredIdx)
    println(filteredNot)

}

还有一些函数通过过滤给定类型的元素来缩小元素类型：

filterIsInstance() 返回给定类型的集合元素。在 List<Any> 上调用 filterIsInstance<T>() 返回 List<T>，因此允许你在其项上调用 T 类型的函数。

fun main() {

    val numbers = listOf(null, 1, "two", 3.0, "four")
    println("All String elements in upper case:")
    numbers.filterIsInstance<String>().forEach {
        println(it.uppercase())
    }

}

filterNotNull() 返回所有非空元素。在 List<T?> 上调用时，filterNotNull() 返回 List<T: Any>，因此允许你将元素视为非空对象。

fun main() {

    val numbers = listOf(null, "one", "two", null)
    numbers.filterNotNull().forEach {
        println(it.length)   // length is unavailable for nullable Strings
    }

}

分区 (Partition)

另一个过滤函数 – partition() – 通过谓词过滤集合，并将不匹配的元素保存在单独的列表中。因此，你将获得一个 Pair 类型的 List 作为返回值：第一个列表包含与谓词匹配的元素，第二个列表包含原始集合中的所有其他元素。

fun main() {

    val numbers = listOf("one", "two", "three", "four")
    val (match, rest) = numbers.partition { it.length > 3 }

    println(match)
    println(rest)

}

测试谓词

最后，有一些函数只是针对集合元素测试谓词：

any() 如果至少有一个元素匹配给定的谓词，则返回 true。
none() 如果没有元素匹配给定的谓词，则返回 true。
all() 如果所有元素都匹配给定的谓词，则返回 true。请注意，当在空集合上使用任何有效的谓词调用 all() 时，它都会返回 true。这种行为在逻辑上被称为_空真 (vacuous truth)_。

fun main() {

    val numbers = listOf("one", "two", "three", "four")

    println(numbers.any { it.endsWith("e") })
    println(numbers.none { it.endsWith("a") })
    println(numbers.all { it.endsWith("e") })

    println(emptyList<Int>().all { it > 5 })   // vacuous truth

}

any() 和 none() 也可以在不使用谓词的情况下使用：在这种情况下，它们只是检查集合是否为空。any() 如果有元素则返回 true，如果没有则返回 false；none() 则相反。

fun main() {

    val numbers = listOf("one", "two", "three", "four")
    val empty = emptyList<String>()

    println(numbers.any())
    println(empty.any())
    
    println(numbers.none())
    println(empty.none())

}

按谓词过滤​

分区 (Partition)​

测试谓词​

按谓词过滤

分区 (Partition)

测试谓词