使用stream将list转成map

2023年1月9日 248次阅读 来源: 熬夜是小狗

jdk1.8 lambda表达式一行便可以将list转成map

其中包括不同的value，以及不同的key及科研重复时解决方案

import java.util.Arrays;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;

import org.junit.Test;

/**
 * Created by Jackielee on 2017
 * @author: lizhilong
 * @date:   2017-11-24 10:13:57   
 * @Copyright: 2017 www.aliyun.com Inc. All rights reserved.
 */
public class DemoListToMap {
    List<Student> list = Arrays.asList(new Student(1, 18, "阿龙", GenderColumn.BOY.getCode()),
                                       new Student(2, 17, "小花", GenderColumn.GIRL.getCode()),
                                       new Student(3, 17, "阿浪", GenderColumn.LADYBOY.getCode()));
    @Test
    public void listToMapByObjectValue(){
        // value 为对象 student -> student jdk1.8返回当前对象
        Map<Integer, Student> map = list.stream().collect(Collectors.toMap(Student::getId, student -> student));
        // 遍历打印结果
        map.forEach((key, value) -> {
            System.out.println("key: " + key + "    value: " + value);
        });
    }
    @Test
    public void listToMapByNameValue(){
        // value 为对象中的属性
        Map<Integer, String> map = list.stream().collect(Collectors.toMap(Student::getId, Student::getName));
        map.forEach((key, value) -> {
            System.out.println("key: " + key + "    value: " + value);
        });
    }
}

value为对象执行结果：

key: 1    value: Student [id=1, age=18, name=阿龙, gender=0]
key: 2    value: Student [id=2, age=17, name=小花, gender=1]
key: 3    value: Student [id=3, age=17, name=阿浪, gender=2]

value为字段执行结果：

key: 1    value: 阿龙
key: 2    value: 小花
key: 3    value: 阿浪

要注意的是map的key必须唯一，所以有可能出现不唯一的时候，就会报错

@Test
    public void listToMapByAgeKey(){
        // value 为对象中的属性
        Map<Integer, String> map = list.stream().collect(Collectors.toMap(Student::getAge, Student::getName));
    }

执行结果：

java.lang.IllegalStateException: Duplicate key 小花

因为age有相同就会报错

此时stream包下Collectors.toMap方法有一个重载方法的参数，这个参数可以传一个合并的函数解决冲突

@Test
    public void listToMapByAgeKey(){
        // value 为对象中的属性
        Map<Integer, String> map = list.stream().collect(
            Collectors.toMap(Student::getAge, Student::getName, (key1, key2) -> key1)
        );
        map.forEach((key, value) -> {
            System.out.println("key: " + key + "    value: " + value);
        });
    }

执行结果：

key: 17    value: 小花
key: 18    value: 阿龙

纯手工操作，欢迎批评指正

    原文作者：熬夜是小狗
    原文地址: https://blog.csdn.net/icannotdebug/article/details/78621705
    本文转自网络文章，转载此文章仅为分享知识，如有侵权，请联系博主进行删除。
点赞