您的位置:

Java Stream转换Map

一、介绍

Java 8中引入了Stream API,提供了丰富的功能,方便我们对集合进行各种操作,其中一个非常常见的操作就是将Stream转换为一个Map。

二、Stream转换为Map的方法

Java 8为了方便我们将Stream转换为Map,提供了两个方法:collect()和toMap()。

三、collect()方法

collect()方法是一个非常强大的方法,可以让我们将Stream中的元素汇总为一个集合,或者进行分组、过滤等操作,并将结果转换为List、Set、Map等类型。

将Stream转换为Map的语法如下:

Map<K, V> map = stream.collect(Collectors.toMap(keyMapper, valueMapper));

其中,keyMapper和valueMapper是两个函数式接口,分别用于将Stream的每个元素映射为Map的key和value。如果key存在重复的情况,会抛出IllegalStateException,可以使用第三个参数mergeFunction来处理重复的key。

下面是一个示例代码:

List<String> list = Arrays.asList("a", "b", "c", "d", "e");
Map<String, Integer> map = list.stream().collect(Collectors.toMap(Function.identity(), String::length));
System.out.println(map); // 输出:{a=1, b=1, c=1, d=1, e=1}

这个例子中,我们将一个List转换为一个String为key,Integer为value的Map,这个Map中所有的value都等于1。

四、toMap()方法

toMap()方法也是用于将Stream转换为Map的方法,不过与collect()方法不同的是,它直接返回一个Map,而不是一个Collector。

toMap()方法的语法如下:

Map<K, V> map = stream.collect(Collectors.toMap(keyMapper, valueMapper, mergeFunction));

其中,mergeFunction是用于处理重复key的情况的函数式接口。如果key重复了,将会调用mergeFunction来处理这个key对应的value。

下面是一个示例代码:

List<String> list = Arrays.asList("apple", "banana", "cherry", "apple");
Map<String, Integer> map = list.stream().collect(Collectors.toMap(Function.identity(), String::length, (v1, v2) -> v1 + v2));
System.out.println(map); // 输出:{apple=10, banana=6, cherry=6}

这个例子中,我们将一个List转换为一个String为key,Integer为value的Map。因为List中有两个“apple”,所以mergeFunction将这两个“apple”的value相加,得到10。

五、结论

Java Stream提供了非常方便的方法将一个Stream转换为Map类型,我们可以使用collect()方法或toMap()方法来实现。在使用的时候,需要注意处理重复key的情况。