使用 Java Stream API 分区数据流并获取不同的集合

本文介绍如何使用 Java Stream API 将数据流按照特定条件进行分区，并将分区后的数据分别收集到不同的集合中。通过 Collectors.partitioningBy 方法，可以高效地将数据流划分为满足条件和不满足条件的两部分，从而避免使用传统的循环方式，简化代码并提高可读性。

使用 partitioningBy 方法进行数据流分区

Java Stream API 提供了 Collectors.partitioningBy 方法，可以根据给定的 Predicate（断言）将数据流划分为两个集合：一个集合包含所有满足 Predicate 的元素，另一个集合包含所有不满足 Predicate 的元素。

partitioningBy 方法返回一个 Map<Boolean, List>，其中 Boolean 类型的键表示元素是否满足 Predicate，List 类型的值分别存储满足和不满足 Predicate 的元素列表。

示例代码：

立即学习“Java免费学习笔记（深入）”；

魔乐社区

天翼云和华为联合打造的AI开发者社区，支持AI模型评测训练、全流程开发应用

102 查看详情

假设我们有一个 ID 列表 myIdList 和一个将 ID 映射到对象的 Map myObjectMap。我们的目标是根据 ID 是否存在于 myObjectMap 中，将 ID 分成两组：存在于 myObjectMap 中的 ID 对应的对象放入一个 List，不存在于 myObjectMap 中的 ID 放入另一个 List。

import java.util.*;import java.util.stream.Collectors;public class StreamPartitioning {    public static void main(String[] args) {        List myIdList = Arrays.asList("a", "b", "c");        Map myObjectMap = new HashMap();        myObjectMap.put("b", "B");        Map<Boolean, List> partitioned = myIdList.stream()                .collect(Collectors.partitioningBy(myObjectMap::containsKey));        List