Java集合Stream类filter的使用

柔光的暖阳◎ 2022-02-09 13:49 583阅读 0赞

# Java集合Stream类filter的使用实例 #

在之前的[Java集合中removeIf的使用][Java_removeIf]中，说明了使用`removeIf`来实现按条件对集合进行过滤。这篇文章同样是使用JDK1.8新加入的`Stream`中`filter`方法来实现同样的效果。并且在实际项目中通常使用`filter`更多。  
和上文一样，首先假设一个场景，你是学校的老师，学生考试成绩太多，有部分学生没有成绩，为了节约时间，现需按照一点规则过滤一下这些学生。**只想查看学生成绩不为null的**。

//人的实体类
    public class Person { 
    
        private String name;//姓名
        private Integer age;//年龄
        private String gender;//性别
        private Integer achievement;//成绩
        
    	...
    	//省略构造方法和getter、setter方法
    	...
    	
    	//重写toString
        @Override
        public String toString() { 
            return "Person{" +
                    "name='" + name + '\'' +
                    ", age=" + age +
                    ", gender='" + gender + '\'' +
                    ", achievement=" + achievement +
                    '}';
        }
    }

这里就不展示使用传统`Iterator`来进行过滤了，有需要做对比的可以参见之前的[Java集合中removeIf的使用][Java_removeIf]。  
使用`Stream`的`filter`进行过滤，只保留成绩不为空的操作：

public static void main(String[] args) { 
       Collection<Person> collection = new ArrayList();
        collection.add(new Person("张三",20,"男",70));
        collection.add(new Person("李四",19,"女",null));
        collection.add(new Person("王五",20,"男",59));
        collection.add(new Person("赵六",18,"男",null));
        collection.add(new Person("田七",20,"女",50));
        Stream<Person> personStream = collection.stream().filter(new Predicate<Person>() { 
            @Override
            public boolean test(Person person) { 
                return null != person.getAchievement();//过滤成绩为null
            }
        });
        collection = personStream.collect(Collectors.toList());//将Stream转化为List
        System.out.println(collection.toString());
    }

运行结果如下：

[Person{name='张三', age=20, gender='男', achievement=70}, Person{name='王五', age=20, gender='男', achievement=59}, Person{name='田七', age=20, gender='女', achievement=50}]

上面的demo没有使用lambda表达式，下面的demo使用lambda来进一步精简代码：

public static void main(String[] args) { 
        Collection<Person> collection = new ArrayList();
        collection.add(new Person("张三",20,"男",70));
        collection.add(new Person("李四",19,"女",null));
        collection.add(new Person("王五",20,"男",59));
        collection.add(new Person("赵六",18,"男",null));
        collection.add(new Person("田七",20,"女",50));
        Stream<Person> personStream1 = collection.stream().filter(p -> p.getAchievement()!=null);//过滤成绩为null
        collection = personStream1.collect(Collectors.toList());//将Stream转化为List
        System.out.println(collection.toString());//结果
    }

效果和不用`lambda`是一样的。  
不过读者在使用`filter`时不要和`removeIf`弄混淆了：

*  removeIf中的test方法返回true代表当前元素会被过滤掉；
 *  filter中的test方法返回true代表当前元素会保留下来。

[Java_removeIf]: https://blog.csdn.net/qq_34292479/article/details/89786154