一、介绍
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的情况。