怎么将一个stream流分成两个流

在 Java 中，将一个 `Stream` 拆分为两个独立的流是一个常见的需求，尤其在处理数据集时需要根据条件或逻辑进行分流操作。虽然 Java 标准库中没有直接提供“拆分”方法，但可以通过多种方式实现这一目标。 ### 使用 `Collectors.partitioningBy` 进行条件拆分 可以使用 `Collectors.partitioningBy` 将流按照某个条件拆分为两个部分。该方法返回一个 `Map<Boolean, List<T>>`，其中键为 `true` 和 `false`，分别表示满足和不满足条件的元素集合。 ```java import java.util.*; import java.util.stream.Collectors; public class StreamSplitExample { public static void main(String[] args) { List<Integer> numbers = Arrays.asList(1, 2, 3, 4, 5, 6); Map<Boolean, List<Integer>> partitioned = numbers.stream() .collect(Collectors.partitioningBy(n -> n % 2 == 0)); System.out.println("Even numbers: " + partitioned.get(true)); // Even numbers: [2, 4, 6] System.out.println("Odd numbers: " + partitioned.get(false)); // Odd numbers: [1, 3, 5] } } ``` ### 使用 `Collectors.groupingBy` 进行多组拆分 如果需要根据多个条件将流拆分成多个子流，可以使用 `Collectors.groupingBy`。它允许根据分类函数将元素分组到不同的列表中。 ```java import java.util.*; import java.util.stream.Collectors; public class StreamGroupingExample { public static void main(String[] args) { List<String> words = Arrays.asList("apple", "banana", "apricot", "blueberry", "avocado"); Map<Integer, List<String>> grouped = words.stream() .collect(Collectors.groupingBy(String::length)); System.out.println("Words grouped by length: " + grouped); // {5=[apple], 6=[banana, apricot], 9=[blueberry], 7=[avocado]} } } ``` ### 使用 `Stream.filter()` 手动创建两个流 也可以通过两次调用 `filter()` 方法来手动创建两个流。这种方式适用于简单的拆分逻辑。 ```java import java.util.*; import java.util.stream.Stream; public class ManualSplitExample { public static void main(String[] args) { List<Integer> numbers = Arrays.asList(1, 2, 3, 4, 5, 6); Stream<Integer> evenStream = numbers.stream().filter(n -> n % 2 == 0); Stream<Integer> oddStream = numbers.stream().filter(n -> n % 2 != 0); System.out.print("Even numbers: "); evenStream.forEach(System.out::print); // 246 System.out.print("\nOdd numbers: "); oddStream.forEach(System.out::print); // 135 } } ``` ### 使用第三方库（如 Vavr 或 jOOλ） 某些第三方库提供了更强大的流处理功能，例如 Vavr 提供了 `Stream#splitAt` 等方法，可以更灵活地拆分流。例如： ```java import io.vavr.collection.Stream; public class VavrSplitExample { public static void main(String[] args) { Stream<Integer> stream = Stream.of(1, 2, 3, 4, 5); io.vavr.collection.Stream.Split<Integer> split = stream.splitAt(2); System.out.println("First part: " + split._1); // First part: Stream(1, 2) System.out.println("Second part: " + split._2); // Second part: Stream(3, 4, 5) } } ``` ### 使用自定义收集器进行复杂拆分 对于更复杂的拆分逻辑，可以编写自定义的 `Collector` 来实现特定的拆分策略。这种方法具有高度灵活性，但实现较为复杂。 ---