java 8 新特性 收集Stream流中的结果

IntStream intStream = Stream.of(1, 2, 3, 4, 5).mapToInt(Integer::intValue);
intStream.filter(n -> n > 3).forEach(System.out::println);
intStream.filter(n -> n > 3).count;
intStream.filter(n -> n > 3).reduce(0, Integer::sum);

对流操作完成之后,如果需要将流的结果保存到数组或集合中,可以收集流中的数据

Stream流中的结果到集合中

Stream流提供 collect 方法,其参数需要一个 java.util.stream.Collector 接口对象来指定收集到哪 种集合中。java.util.stream.Collectors 类提供一些方法,可以作为 Collector`接口的实例:

  1. public static Collector> toList() :转换为 List 集合。
  2. public static Collector> toSet() :转换为 Set 集合。

下面是这两个方法的基本使用代码:

// 将流中数据收集到集合中
@Test
public void testStreamToCollection() {
Stream<String> stream = Stream.of("aa", "bb", "cc");
    // List<String> list = stream.collect(Collectors.toList());
    // Set<String> set = stream.collect(Collectors.toSet());
    ArrayList<String> arrayList = stream.collect(Collectors.toCollection(ArrayList::new));
    HashSet<String> hashSet = stream.collect(Collectors.toCollection(HashSet::new));
}

Stream流中的结果到数组中

Stream提供 toArray 方法来将结果放到一个数组中,返回值类型是Object[]的:

Object[] toArray();

其使用场景如:

@Test
public void testStreamToArray() {
    Stream<String> stream = Stream.of("aa", "bb", "cc");
    // Object[] objects = stream.toArray();
    // for (Object obj : objects) {
    // System.out.println();
    // }
    
    String[] strings = stream.toArray(String[]::new);
    for (String str : strings) {
         System.out.println(str);
    }
}

对流中数据进行聚合计算

当我们使用Stream流处理数据后,可以像数据库的聚合函数一样对某个字段进行操作。比如获取最大值,获取最小 值,求总和,平均值,统计数量。

@Test
public void testStreamToOther() {
    Stream<Student> studentStream = Stream.of(
                    new Student("赵丽颖", 58, 95),
                    new Student("杨颖", 56, 88),
                    new Student("迪丽热巴", 56, 99),
                    new Student("柳岩", 52, 77));
    // 获取最大值
     Optional<Student> collect = studentStream.collect(Collectors.maxBy((o1, o2) ->
        o1.getSocre() - o2.getSocre()));
    
    // 获取最小值
     Optional<Student> collect = studentStream.collect(Collectors.minBy((o1, o2) ->
        o1.getSocre() - o2.getSocre()));
    // System.out.println(collect.get());
    
    
    // 求总和
     int sumAge = studentStream.collect(Collectors.summingInt(s -> s.getAge()));
     System.out.println("sumAge = " + sumAge);
    
     //平均值
     double avgScore = studentStream.collect(Collectors.averagingInt(s -> s.getSocre()));
     System.out.println("avgScore = " + avgScore);
    
    // 统计数量
     Long count = studentStream.collect(Collectors.counting());
     System.out.println("count = " + count);
}

对流中数据进行分组

当我们使用Stream流处理数据后,可以根据某个属性将数据分组:

// 分组
@Test
public void testGroup() {
    Stream<Student> studentStream = Stream.of(
        new Student("赵丽颖", 52, 95),
        new Student("杨颖", 56, 88),
        new Student("迪丽热巴", 56, 55),
        new Student("柳岩", 52, 33));
    // Map<Integer, List<Student>> map =
    studentStream.collect(Collectors.groupingBy(Student::getAge));
   
    // 将分数大于60的分为一组,小于60分成另一组
    Map<String, List<Student>> map = studentStream.collect(Collectors.groupingBy((s) ->
        {
            if (s.getSocre() > 60) {
                 return "及格";
             } else {
                 return "不及格";
                }
            }));
    
    map.forEach((k, v) -> {
          System.out.println(k + "::" + v);
       });


}

效果:

不及格::[Student{name='迪丽热巴', age=56, socre=55}, Student{name='柳岩', age=52, socre=33}]
及格::[Student{name='赵丽颖', age=52, socre=95}, Student{name='杨颖', age=56, socre=88}]

对流中数据进行多级分组

还可以对数据进行多级分组:

// 多级分组
@Test
public void testCustomGroup() {
    Stream<Student> studentStream = Stream.of(
        new Student("赵丽颖", 52, 95),
        new Student("杨颖", 56, 88),
        new Student("迪丽热巴", 56, 99),
        new Student("柳岩", 52, 77));
    Map<Integer, Map<String, List<Student>>> map =
    studentStream.collect(Collectors.groupingBy(s -> s.getAge(), Collectors.groupingBy(s -> {
            if (s.getSocre() >= 90) {
             return "优秀";
                } else if (s.getSocre() >= 80 && s.getSocre() < 90) {
                return "良好";
             } else if (s.getSocre() >= 80 && s.getSocre() < 80) {
                return "及格";
                } else {
                return "不及格";
              }
            })));
    
    map.forEach((k, v) -> {
         System.out.println(k + " == " + v);
    });
}

效果:

52 == {不及格=[Student{name='柳岩', age=52, socre=77}], 优秀=[Student{name='赵丽颖', age=52,socre=95}]}56 == {优秀=[Student{name='迪丽热巴', age=56, socre=99}], 良好=[Student{name='杨颖', age=56,socre=88}]}

对流中数据进行分区

Collectors.partitioningBy 会根据值是否为true,把集合分割为两个列表,一个true列表,一个false列表。

image-20211201143755915.png
// 分区@Testpublic void testPartition() {    Stream<Student> studentStream = Stream.of(        new Student("赵丽颖", 52, 95),        new Student("杨颖", 56, 88),        new Student("迪丽热巴", 56, 99),        new Student("柳岩", 52, 77));    // partitioningBy会根据值是否为true,把集合分割为两个列表,一个true列表,一个false列表。    Map<Boolean, List<Student>> map = studentStream.collect(Collectors.partitioningBy(s ->                s.getSocre() > 90));       map.forEach((k, v) -> {            System.out.println(k + " == " + v);    });}

效果:

false == [Student{name='杨颖', age=56, socre=88}, Student{name='柳岩', age=52, socre=77}]true == [Student{name='赵丽颖', age=52, socre=95}, Student{name='迪丽热巴', age=56, socre=99}]

对流中数据进行拼接

// 拼接@Testpublic void testJoining() {Stream<Student> studentStream = Stream.of(    new Student("赵丽颖", 52, 95),    new Student("杨颖", 56, 88),    new Student("迪丽热巴", 56, 99),    new Student("柳岩", 52, 77));    String collect = studentStream            .map(Student::getName)            .collect(Collectors.joining(">_<", "^_^", "^v^"));    System.out.println(collect);}

效果:

^_^赵丽颖>_<杨颖>_<迪丽热巴>_<柳岩^v^

小结

收集Stream流中的结果

  1. 到集合中: Collectors.toList()/Collectors.toSet()/Collectors.toCollection()
  2. 到数组中: toArray()/toArray(int[]::new)

聚合计算:

Collectors.maxBy/Collectors.minBy/Collectors.counting/Collectors.summingInt/Collectors.averagingInt

  1. 分组: Collectors.groupingBy
  2. 分区: Collectors.partitionBy
  3. 拼接: Collectors.joingin
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 202,980评论 5 476
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 85,178评论 2 380
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 149,868评论 0 336
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 54,498评论 1 273
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 63,492评论 5 364
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 48,521评论 1 281
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 37,910评论 3 395
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,569评论 0 256
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 40,793评论 1 296
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,559评论 2 319
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,639评论 1 329
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,342评论 4 318
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 38,931评论 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,904评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,144评论 1 259
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 42,833评论 2 349
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,350评论 2 342

推荐阅读更多精彩内容