当前位置：首页 > article >正文

Java流Stream使用详解（下）

article 2024/11/18 1:29:36

收集方法collect

收集到List、Set中比较简单，收集到Map中是重点，有点复杂，需要好好的理解。注意收集的Map中的数据键不能重复，否则会抛出异常。

import java.util.*;
import java.util.function.Function;
import java.util.stream.Collectors;

public class StreamDemo10 {
    public static void main(String[] args) {
        /*
        collect(Collector collector)    收集流中的数据，放到集合中（List Set Map）
        */

        // 创建数组
        ArrayList<String> list =new ArrayList<>();
        // 添加元素
        Collections.addAll(list,"张无忌-男-15","周芷若-女-14","赵敏-女-13","张强-男-20",
                                        "张三丰-男-100","张翠山-男-40","张良-男-35","王二麻子-男-37","谢广坤-男-41");
        //收集List集合当中
        //需求：
        //我要把所有的男性收集起来
        List<String> newList = list.stream().filter(s -> "男".equals(s.split("-")[1])).collect(Collectors.toList());

        // 收集到Set集合中
        // 需求：
        // 我要把所有的男性收集起来
        Set<String> newList2 = list.stream().filter(s -> "男".equals(s.split("-")[1]))
                .collect(Collectors.toSet());


        Map<String, Integer> map = list.stream().filter(s -> "男".equals(s.split("-")[1]))
                /*
                 *  要用map收集数据注意键不能重复，否则会抛出异常
                 *
                 *  toMap :参数一表示键的生成规则
                 *         参数二表示值的生成规则
                 *
                 *  参数一：
                 *          Function泛型一：表示流中每一个数据的类型
                 *                   泛型二：表示Map集合中键的数据类型
                 *           方法apply形参：依次表示流里面的每一个数据
                 *                  方法体：生成键的代码
                 *                  返回值：已经生成的键
                 *
                 *  参数二：
                 *           Function泛型一：表示流中每一个数据的类型
                 *           泛型二：表示Map集合中值的数据类型
                 *
                 *
                 *           方法apply形参：依次表示流里面的每一个数据
                 *                  方法体：生成值的代码
                 *                  返回值：已经生成的值
                 *
                 *
                 */

                .collect(Collectors.toMap(new Function<String, String>() {
                    @Override
                    public String apply(String s) {
                        return s.split("-")[0];
                    }
                }, new Function<String, Integer>() {
                    @Override
                    public Integer apply(String s) {
                        return Integer.valueOf(s.split("-")[2]);
                    }
                }));
        //System.out.println(map);

        // 改成lambda表达式
        Map<String, Integer> map2 = list.stream().filter(s -> "男".equals(s.split("-")[1]))
                .collect(Collectors.toMap(s -> s.split("-")[0], s -> Integer.valueOf(s.split("-")[2])));
        System.out.println(map2);
    }
}