kotlin 将一个list按条件分为两个list(partition )
前言
在安卓开发过程中,我们经常需要将一个列表按照特定条件拆分为两个子列表。随着对语言的深入理解,我发现了一些更高效、更简洁的实现方式,现在与大家分享。
实现
传统Java实现
假设我们有以下列表需要处理:
List<String> list = new ArrayList<>();
list.add("1");
list.add("2");
list.add("1");
list.add("2");
使用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优化
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实现
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实现
val (list1, list2) = list.partition { it == "1" }