问题背景：传统数组过滤的低效实践
在java开发中，我们经常会遇到需要从一个现有数组中根据特定条件（例如，大于某个阈值）筛选出部分元素，并将这些元素放入一个新的数组中的场景。一种常见的、但效率较低的实现方式是使用两个独立的循环：第一个循环用于统计满足条件的元素数量，以便为新数组分配正确的内存空间；第二个循环则遍历原始数组，将满足条件的元素逐一复制到新数组中。
例如，以下代码展示了这种双循环方法：
public int[] getValuesAboveThreshold(int[] data, int threshold) {
// 第一次循环：统计满足条件的元素数量
int counter = 0;
for (int i = 0; i < data.length; i++) {
if (data[i] > threshold) {
counter++;
}
}
// 根据统计结果创建新数组
int[] thresholdArray = new int[counter];
// 第二次循环：将满足条件的元素填充到新数组
int count = 0;
for (int i = 0; i < data.length; i++) {
if (data[i] > threshold) {
thresholdArray[count] = data[i];
count++;
}
}
return thresholdArray;
}这种方法虽然功能上可行，但存在以下缺点：
- 代码冗余： 两个循环的逻辑相似，导致代码重复。
- 效率低下： 需要两次完整遍历原始数组，对于大型数组而言，会增加不必要的计算开销。
- 可读性差： 意图不够明确，需要仔细阅读才能理解其完整逻辑。
Stream API：现代Java的高效解决方案
Java 8引入的Stream API提供了一种更函数式、更声明式的方法来处理集合数据。通过Stream，我们可以以一种链式操作的方式，对数据进行过滤、映射、排序等多种处理，从而显著提高代码的简洁性和可读性。对于数组过滤这种场景，Stream API提供了一种优雅且高效的解决方案，避免了手动管理循环和数组大小的复杂性。
使用Stream API过滤数组
利用Stream API过滤数组并生成新数组的核心在于以下三个步骤：
立即学习“Java免费学习笔记（深入）”；
- 将数组转换为Stream： 使用Arrays.stream()方法将原始数组转换为一个特定类型的Stream（例如，int[]转换为IntStream）。
- 应用过滤条件： 使用filter()中间操作，传入一个Lambda表达式作为谓词（Predicate），筛选出满足条件的元素。
- 将Stream转换回数组： 使用toArray()终端操作，将经过过滤的Stream中的元素收集到一个新的数组中。
以下是使用Stream API实现上述数组过滤功能的示例代码：
import java.util.Arrays;
public class ArrayFilteringExample {
public static void main(String[] args) {
int threshold = 4;
int[] data = new int[]{1, 2, 3, 4, 5, 6, 7, 8, 9};
// 调用Stream API方法进行过滤
int[] filteredArray = getValuesAboveThreshold(data, threshold);
System.out.println("原始数组: " + Arrays.toString(data));
System.out.println("阈值: " + threshold);
System.out.println("过滤后的数组: " + Arrays.toString(filteredArray));
}
/**
* 使用Stream API从数组中筛选出大于指定阈值的元素。
*
* @param originalArray 原始整数数组。
* @param threshold 过滤阈值。
* @return 包含所有大于阈值元素的整数数组。
*/
private static int[] getValuesAboveThreshold(int[] originalArray, int threshold) {
return Arrays.stream(originalArray) // 1. 将原始数组转换为IntStream
.filter(val -> val > threshold) // 2. 过滤出大于阈值的元素
.toArray(); // 3. 将过滤后的元素收集到新的int数组中
}
}代码解析：
- Arrays.stream(originalArray): 这是一个工厂方法，用于将一个int[]数组转换为一个IntStream。IntStream是Java为原始类型int提供的特殊Stream，避免了自动装箱/拆箱的性能开销。
- .filter(val -> val > threshold): 这是一个中间操作。它接收一个IntPredicate（一个接受int并返回boolean的函数式接口）作为参数。这里使用Lambda表达式val -> val > threshold，表示只有当元素val大于threshold时，才会被保留在Stream中，并传递给下一个操作。
- .toArray(): 这是一个终端操作。它会触发Stream的执行，并将Stream中剩余的所有元素收集到一个新的int[]数组中并返回。
Stream API的优势与注意事项
优势
- 简洁性与可读性： Stream API以声明式风格编写，代码更加精炼，能够清晰地表达“做什么”而不是“如何做”，提高了代码的可读性和维护性。
- 效率提升： 尽管对于极小的数组，传统循环可能略快，但对于中大型数组，Stream API的内部优化（如延迟执行、短路操作）以及并行流（parallelStream()）的引入，可以显著提升处理效率。
- 函数式编程范式： 鼓励使用不可变数据和无副作用的操作，有助于编写更健壮、更易于测试的代码。
- 链式操作： 允许将多个操作（如过滤、映射、排序）以链式调用的方式组合起来，形成一个清晰的数据处理管道。
注意事项
- 理解Stream的生命周期： Stream是一次性的，一旦经过终端操作（如toArray()、forEach()），就不能再次使用。如果需要重新处理数据，需要重新创建Stream。
- 性能考量： 并非所有场景都适合使用Stream。对于非常简单的循环操作，传统for循环可能在性能上略有优势。但在涉及复杂数据转换、过滤或需要并行处理时，Stream API的优势更为明显。
- 原始类型Stream： 对于int、long、double等原始类型，应优先使用IntStream、LongStream、DoubleStream，以避免自动装箱/拆箱带来的性能损耗。
- 调试： Stream管道的调试可能比传统循环稍微复杂一些，可以使用peek()中间操作来观察Stream在不同阶段的数据状态。
总结
通过本文的介绍，我们了解了在Java中高效过滤数组元素并生成新数组的两种方法。相较于传统的双循环方法，Java 8引入的Stream API提供了一种更为现代、简洁且高效的解决方案。通过Arrays.stream().filter().toArray()的链式操作，开发者可以编写出更具可读性、更易于维护的代码，同时也能利用Stream API的内部优化和并行处理能力，提升数据处理的性能。在实际开发中，应优先考虑使用Stream API来处理数组和集合的转换与过滤任务。
