在java编程中，我们经常会遇到需要从一个现有数组中筛选出符合特定条件的元素，并将这些元素收集到一个新数组中的场景。例如，从一个整数数组中找出所有大于某个阈值的值。传统的实现方式通常涉及两个循环：第一个循环用于计算符合条件的元素数量，以便为新数组分配正确的内存大小；第二个循环则遍历原始数组，将符合条件的元素逐一填充到新数组中。这种方法虽然可行，但代码显得冗余且不够简洁。
传统数组过滤方法的挑战
考虑以下使用传统双循环过滤数组的示例：
public class ArrayFilterLegacy {
// 假设存在一个名为 data 的类成员数组
// private int[] data = {1, 2, 3, 4, 5, 6, 7, 8, 9};
// 为方便演示，我们将其作为参数传入方法
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;
}
public static void main(String[] args) {
int threshold = 4;
int[] data = new int[]{1, 2, 3, 4, 5, 6, 7, 8, 9};
ArrayFilterLegacy filter = new ArrayFilterLegacy();
int[] filteredArray = filter.getValuesAboveThreshold(data, threshold);
System.out.println("传统方法过滤结果: " + java.util.Arrays.toString(filteredArray)); // 输出: [5, 6, 7, 8, 9]
}
}这段代码清晰地展示了传统方法的两阶段处理：首先预扫描以确定新数组大小，然后再次扫描以填充数据。这种模式在功能上没有问题，但在代码的简洁性和可读性方面有提升空间。
利用Java Stream API进行高效过滤
Java 8引入的Stream API提供了一种更现代、更函数式的方法来处理集合和数组数据。它允许我们以声明式的方式表达数据处理管道，从而使代码更加简洁、易读且易于维护。
使用Stream API过滤数组元素并生成新数组的核心步骤如下：
立即学习“Java免费学习笔记（深入）”；
-
将数组转换为Stream： 使用 Arrays.stream() 方法将原始数组转换为一个 IntStream (对于 int[]) 或 Stream
(对于对象数组)。 - 应用过滤条件： 使用 filter() 方法，传入一个 Predicate（通常是Lambda表达式），来定义筛选条件。只有满足条件的元素才能通过此阶段。
- 收集结果： 使用 toArray() 方法将过滤后的Stream元素收集回一个新的数组。
下面是使用Stream API实现相同功能的示例代码：
import java.util.Arrays;
public class ArrayFilterStream {
/**
* 使用Java Stream API过滤数组中大于指定阈值的元素。
*
* @param originalArray 原始整数数组。
* @param threshold 过滤阈值。
* @return 包含所有大于阈值的元素的新数组。
*/
private static int[] getValuesAboveThreshold(int[] originalArray, int threshold) {
return Arrays.stream(originalArray) // 1. 将int[]转换为IntStream
.filter(val -> val > threshold) // 2. 过滤出大于阈值的元素
.toArray(); // 3. 将过滤后的元素收集回一个新的int[]
}
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("Stream API过滤结果: " + Arrays.toString(filteredArray)); // 输出: [5, 6, 7, 8, 9]
}
}代码解析
- Arrays.stream(originalArray): 这一步将一个 int 数组转换成一个 IntStream。IntStream 是专门用于处理基本类型 int 的Stream，相比 Stream
在性能上更优。 - .filter(val -> val > threshold): 这是Stream管道中的核心过滤操作。filter() 方法接收一个 IntPredicate 函数式接口的实例，这里我们使用Lambda表达式 val -> val > threshold 来实现。这个Lambda表达式为Stream中的每个元素 val 执行一个条件检查，只有当 val 大于 threshold 时，该元素才能继续流向下一个操作。
- .toArray(): 这是Stream管道的终止操作，它将Stream中剩余的所有元素收集到一个新的 int 数组中并返回。这个方法会自动处理新数组的大小分配，无需预先计数。
Stream API的优势
使用Stream API进行数组过滤带来了多方面的优势：
- 简洁性与可读性： Stream管道以链式调用的方式清晰地表达了数据处理的逻辑，将“做什么”而非“如何做”放在首位，代码量显著减少。
- 函数式编程风格： 鼓励使用不可变数据和纯函数，减少副作用，使代码更易于理解和测试。
- 易于并行化： 对于大规模数据，Stream API可以非常容易地通过 parallelStream() 或 parallel() 方法实现并行处理，从而提高执行效率（尽管对于本例这样的小数组，并行化可能不会带来明显优势，甚至可能引入开销）。
- 避免中间变量： 无需像传统方法那样手动管理 counter 和 count 等中间变量。
注意事项
-
原始类型Stream： 对于 int[]、long[]、double[] 等基本类型数组，应优先使用 Arrays.stream() 返回的 IntStream、LongStream、DoubleStream，而不是将其装箱为 Stream
等，以避免不必要的装箱/拆箱开销，提高性能。 - 终止操作： Stream管道必须以一个终止操作（如 toArray()、collect()、forEach() 等）结束，否则Stream不会执行任何操作。
- Stream的一次性使用： Stream是惰性求值的，并且只能被消费一次。一旦Stream的终止操作被执行，该Stream就不能再被重复使用。
总结
Java Stream API为数组和集合的数据处理提供了强大而优雅的解决方案。通过链式调用 stream()、filter() 和 toArray() 等方法，我们可以以极高的效率和可读性完成数据筛选任务，避免了传统双循环的冗余。掌握Stream API是现代Java开发中不可或缺的技能，它能帮助开发者编写出更简洁、更具表达力的代码。在处理数据转换和过滤的场景中，优先考虑使用Stream API，将显著提升开发效率和代码质量。
