使用Java Stream API高效过滤数组元素并生成新数组

在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免费学习笔记（深入）”；

豆绘AI

豆绘AI是国内领先的AI绘图与设计平台，支持照片、设计、绘画的一键生成。

485

查看详情

1. 将数组转换为Stream： 使用 Arrays.stream() 方法将原始数组转换为一个 IntStream (对于 int[]) 或 Stream<T> (对于对象数组)。
2. 应用过滤条件： 使用 filter() 方法，传入一个 Predicate（通常是Lambda表达式），来定义筛选条件。只有满足条件的元素才能通过此阶段。
3. 收集结果： 使用 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<Integer> 在性能上更优。
• .filter(val -> val > threshold): 这是Stream管道中的核心过滤操作。filter() 方法接收一个 IntPredicate 函数式接口的实例，这里我们使用Lambda表达式 val -> val > threshold 来实现。这个Lambda表达式为Stream中的每个元素 val 执行一个条件检查，只有当 val 大于 threshold 时，该元素才能继续流向下一个操作。
• .toArray(): 这是Stream管道的终止操作，它将Stream中剩余的所有元素收集到一个新的 int 数组中并返回。这个方法会自动处理新数组的大小分配，无需预先计数。

Stream API的优势

使用Stream API进行数组过滤带来了多方面的优势：

1. 简洁性与可读性： Stream管道以链式调用的方式清晰地表达了数据处理的逻辑，将“做什么”而非“如何做”放在首位，代码量显著减少。
2. 函数式编程风格： 鼓励使用不可变数据和纯函数，减少副作用，使代码更易于理解和测试。
3. 易于并行化： 对于大规模数据，Stream API可以非常容易地通过 parallelStream() 或 parallel() 方法实现并行处理，从而提高执行效率（尽管对于本例这样的小数组，并行化可能不会带来明显优势，甚至可能引入开销）。
4. 避免中间变量： 无需像传统方法那样手动管理 counter 和 count 等中间变量。

注意事项

• 原始类型Stream： 对于 int[]、long[]、double[] 等基本类型数组，应优先使用 Arrays.stream() 返回的 IntStream、LongStream、DoubleStream，而不是将其装箱为 Stream<Integer> 等，以避免不必要的装箱/拆箱开销，提高性能。
• 终止操作： Stream管道必须以一个终止操作（如 toArray()、collect()、forEach() 等）结束，否则Stream不会执行任何操作。
• Stream的一次性使用： Stream是惰性求值的，并且只能被消费一次。一旦Stream的终止操作被执行，该Stream就不能再被重复使用。

总结

Java Stream API为数组和集合的数据处理提供了强大而优雅的解决方案。通过链式调用 stream()、filter() 和 toArray() 等方法，我们可以以极高的效率和可读性完成数据筛选任务，避免了传统双循环的冗余。掌握Stream API是现代Java开发中不可或缺的技能，它能帮助开发者编写出更简洁、更具表达力的代码。在处理数据转换和过滤的场景中，优先考虑使用Stream API，将显著提升开发效率和代码质量。

以上就是使用Java Stream API高效过滤数组元素并生成新数组的详细内容，更多请关注php中文网其它相关文章！