kotlin 将一个list按条件分为两个list（partition ）

我怀里的猫2025-05-21 19:05

前言

在安卓开发过程中，我们经常需要将一个列表按照特定条件拆分为两个子列表。随着对语言的深入理解，我发现了一些更高效、更简洁的实现方式，现在与大家分享。

实现

传统Java实现

假设我们有以下列表需要处理：

java 复制代码

List<String> list = new ArrayList<>();
list.add("1");
list.add("2");
list.add("1");
list.add("2");

使用Java进行条件拆分

java 复制代码

List<String> list1 = new ArrayList<>();
List<String> list2 = new ArrayList<>();
list.forEach(item -> {
    if (Objects.equals(item, "1")) {
        list1.add(item);
    } else {
        list2.add(item);
    }
});

使用Java Stream API优化

java 复制代码

List<String> list1 = list.stream()
                        .filter(item -> Objects.equals(item, "1"))
                        .collect(Collectors.toList());
List<String> list2 = list.stream()
                        .filter(item -> !Objects.equals(item, "1"))
                        .collect(Collectors.toList());

Kotlin实现

kotlin 复制代码

val list: MutableList<String> = ArrayList()
list.add("1")
list.add("2")
list.add("1")
list.add("2")

val list1: List<String> = list.filter { it == "1" }
val list2: List<String> = list.filter { it != "1" }

更简洁的Kotlin实现

kotlin 复制代码

val (list1, list2) = list.partition { it == "1" }