您的位置:首页 > 编程语言 > Java开发

使用stream将list转成map

2017-11-24 11:12 183 查看
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: 阿龙


纯手工操作,欢迎批评指正
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息