【Java】List按照某个属性进行分组，形成二维List,相同的类型放在同一个集合

果壳~

于 2024-09-04 11:02:29 发布

阅读量132

点赞数 3

文章标签： java list windows

本文链接：https://blog.csdn.net/Passer_hua/article/details/141887881

版权

@Data
public class User {

    private String name;

    private String type;

    public User(String name, String type) {
        this.name = name;
        this.type = type;
    }
}

public class TestGroupList {

    @Test
    public void  testGroupList() {
        List<User> userList = new ArrayList<>();
        userList.add(new User("张三","1"));
        userList.add(new User("李四","2"));
        userList.add(new User("王五","3"));
        userList.add(new User("赵六","1"));
        userList.add(new User("钱七","2"));
        userList.add(new User("孙八","3"));
        
        List<List<User>> collect = groupList(userList);
        System.out.println(collect);
    }

    public static List<List<User>> groupList(List<User> userList) {
        Map<String, List<User>> collect = userList.stream().collect(Collectors.groupingBy(User::getType));
        return new ArrayList<>(collect.values());
    }
    
}