当前位置: 代码迷 >> 综合 >> java8 stream().map().collect()的Collectors.toList()、Collectors.toMap()、Collectors.groupingBy()的用法
  详细解决方案

java8 stream().map().collect()的Collectors.toList()、Collectors.toMap()、Collectors.groupingBy()的用法

热度:9   发布时间:2023-11-01 09:43:38.0

一、Collectors.toList()

??现在有个集合:

List<User> users = getUserList();

??现在需要将这些user的id提取出来。这个很简单,for循环嘛,谁不会啊(不会吧不会吧,不会有人还不会用for循环的吧)。

List<Long> idList = new ArrayList<Long>();for(int i = 0; i < users.size(); i++){idList.add(users.get(i).getId());}

??然而Java8有个更简便的方法,一行代码搞定:

List<Long> idList = users.stream().map(User::getId).collect(Collectors.toList());

其中有个User::getId,这个其实就是调用User类的getId()方法。

??再举个例子,将集合里的元素由小写变大写:

List<String> list = Arrays.asList("a", "b", "c", "d");List<String> results = list.stream().map(String::toUpperCase).collect(Collectors.toList());System.out.println(results); //{A, B, C, D}

二、Collectors.toMap()

??Collectors.toMap(),一般用于将一个List转换为Map。常见用法:

list.stream().collect(Collectors.toMap(Function keyMapper, Function valueMapper))

在这里插入图片描述

??可以接收2个、3个、4个参数,但是我一般只用2个的或者3个的就已经足够了。这里我也就只讲一个前两个用法,也就是2个参数的和3个参数的用法。

??第一个参数,用于指定key的Function。
??第二个参数,用于指定value的Function。
??第三个参数,若在转换中,出现多个相同的key,如何进行合并的Function。

1. 两个参数的用法

??现在有个User对象:

@Getter
@Setter
public class User{
    private Long id;private String name;private Integer age;public User(Long id, String name, Integer age) {
    this.id = id;this.name = name;this.age = age;}
}

??然后进行以下操作:

List<User> userList = new ArrayList<>();
userList.add(new User(1, "张三", 18));
userList.add(new User(2, "李四", 19));
userList.add(new User(3, "王五", 18));
//将userList转化为key为id,value为User对象的map
Map<Long, User> map = userList.stream().collect(Collectors.toMap(User::getId, p -> p));

??Map<Long, User> map = userList.stream().collect(Collectors.toMap(User::getId, p -> p));这一步就是将userList 转换为key为id,value为User对象的map。

User::getId ===》 User对象的getId方法
p -> p ===》就是进来的是什么,最终就是什么,这里就是进来的是User对象,出去的也就是User对象

??而这时map里的(模拟)值是:

{
    1: User(1, "张三", 18)2: User(2, "李四", 19)3: User(3, "王五", 18)
}

??还可以换一下:

Map<Long, String> map = userList.stream().collect(Collectors.toMap(User::getId, User::getName));

??这个获取的就是key为id,value为name的map了。

2. 三个参数的用法

??还是沿用上面那个例子,如果这个时候你想获取key是age,value是name的map呢?如果你还是沿用上面的方法,就会出问题了,因为有两个age是 18 的数据,也就是存在重复的key,会直接报错,想不报错的话,就可以利用第三个参数了。

Map<Integer, String> map = userList.stream().collect(Collectors.toMap(User::getAge, User::getName, (a, b) -> b));

??(a, b) -> b的意思就是,如果存在重复的,永远取后面一个

??这时,map里的值就是:

{
    18: "王五"19: "李四"
}

三、Collectors.groupingBy()

??还是沿用上面那个例子。当你想获取key是age的map,又不想覆盖掉重复项数据,这个时候就可以用 Collectors.groupingBy 了。

Map<Integer, List<User>> map = userList.stream().collect(Collectors.groupingBy(User::getAge));

??可以看到,这次的返回值变成了 Map<Integer, List> 了,也就是说,变成了key是age,value是User对象的集合了。这时,map里的值就变成了:

{
    18: [User(1, "张三", 18), User(3, "王五", 18)]19: [User(2, "李四", 19)]
}

以上就是这三种Collectors的用法了,如果你有什么疑点,或者是觉得博主有哪里写的不对的,欢迎在下方留言讨论哦~

  相关解决方案