您的位置:

Java 8中Stream如何将数据转化为Map

Java 8中引入了新的Stream API,它具有流式计算的特性,可以对集合、数组等数据进行处理和计算。其中,Stream的toMap()方法可以将流中的元素转化为Map,本文将从以下方面对如何使用Java 8的Stream将数据转化为Map进行详细阐述。

一、创建HashMap并初始化

转化为Map前,需要创建Map对象并初始化,Java 8中提供了一个静态方法可以创建Map,即Map.of()。但使用Map.of()方法有一个前提,它只适用于小型Map的初始化,如果需要初始化大型Map或一个Map中有多个相同的key,需要使用HashMap来初始化。 使用HashMap初始化,可以先创建Map对象,再使用put()方法为Map添加元素:
Map<String, String> map = new HashMap<>();
map.put("key1", "value1");
map.put("key2", "value2");

二、使用Stream将List转化为Map

使用Java 8的Stream将List转化为Map时,需要指定key和value,可以通过lambda表达式来完成:
List<Person> personList = new ArrayList<>();
personList.add(new Person("Tom", 20));
personList.add(new Person("Jerry", 22));
Map<String, Integer> nameToAgeMap = personList.stream().collect(Collectors.toMap(Person::getName, Person::getAge));
上面的代码中,创建了一个Person类,包含姓名和年龄两个属性,然后将Person对象放入List中。使用stream()方法生成Stream对象,再使用collect()方法并指定toMap()方法,根据getName()方法生成key,根据getAge()方法生成value,最终得到Map<String, Integer>类型的nameToAgeMap。

三、使用Stream将List转化为多个value的Map

如果需要将List转化为一个key对应多个value的Map,可以使用toMap()的第三个参数实现:
List<Person> personList = new ArrayList<>();
personList.add(new Person("Tom", 20));
personList.add(new Person("Jerry", 20));
Map<String, List<Person>> ageToPersonListMap = personList.stream().collect(Collectors.toMap(Person::getAge, 
    p -> new ArrayList<>(Arrays.asList(p)), (oldList, newList) -> {
    oldList.addAll(newList);
    return oldList;
}));
在toMap()方法中,除了getKeyMapper()和getValueMapper()外,还需要提供一个函数来处理key重复(多个对象的key值相同)的情况,上述代码中,通过第三个参数解决问题。第三个参数为一个函数式接口,用于合并两个值,返回的是合并后的值,这里使用了lambda表达式来完成。

四、使用Stream将数组转化为Map

使用Java 8的Stream将数组转化为Map时,需要先将数组转化为Stream对象:
String[] words = {"hello", "world", "i", "am", "a", "bot"};
Map<String, Integer> wordLengthMap = Arrays.stream(words).collect(Collectors.toMap(
    w -> w, w -> w.length(), (oldValue, newValue) -> oldValue, HashMap::new));
上述代码中,使用Arrays.stream()方法将数组转化为Stream对象,再使用toMap()方法,key和value分别为数组元素和元素的长度。第三个参数同样为处理key重复情况的函数,使用HashMap作为返回的Map类型。

五、使用Stream将List转化为Map,value为对象

假设有一个List,需要将List中的某一个属性作为key,整个对象作为value生成一个Map:
List<Person> personList = new ArrayList<>();
personList.add(new Person("Tom", 20));
personList.add(new Person("Jerry", 22));
Map<String, Person> nameToPersonMap = personList.stream().collect(Collectors.toMap(Person::getName, p -> p));
上述代码中,使用toMap()方法时,getValueMapper()使用lambda表达式直接将整个Person对象作为value,当然也可以指定一个返回值为Person类型的方法。

结束语

通过Java 8中的Stream API,并结合lambda表达式等特性,可以方便地将List和数组等数据转化为Map,极大地提高了编程效率,使代码更加简洁、优雅。本文介绍了5种将数据转化为Map的方法,可以根据实际需求选择使用。