使用Java 8中的Lambda将流收集到HashMap中

Question

我有一个HashMap,我需要使用一些函数进行过滤:

HashMap<Set<Integer>, Double> container
Map.Entry<Set<Integer>, Double> map = container.entrySet()
            .stream()
            .filter(k -> k.getKey().size() == size)

对于size = 2,以下内容应该有效:

containerBeforeFilter = {1,2,3} -> 1.5, {1,2} -> 1.3
containerAfterFilter = {1,2} -> 1.3

在过滤器中应用函数后,我想再次将结果收集到HashMap中.但是,当我尝试应用此处建议的方法时,我会收到非法声明.

因此,在过滤器之后应用的以下语句是非法的:

.collect(Collectors.toMap((entry) -> entry.getKey(), (entry) -> entry.getValue()));

收集未更改的地图值的正确方法是什么,唯一的标准是满足某些关键？

UPDATE

上面代码中的错误是变量的类型map.它应该是Map而不是Map.Entry.

所以功能代码是:

Map<Set<Integer>, Double> map = container.entrySet()
            .stream()
            .filter(k -> k.getKey().size() == size)
            .collect(Collectors.toMap(entry -> entry.getKey(), entry -> entry.getValue()));

Answer 1

有一个更好的解决方案

toMap(Function<? super T, ? extends K> keyMapper,
                            Function<? super T, ? extends U> valueMapper,
                            BinaryOperator<U> mergeFunction,
                            Supplier<M> mapSupplier)

您可以这样使用它：

HashMap<Set<Integer>, Double> map = container.entrySet()
    .stream()
    .filter(k -> k.getKey().size() == size)
    .collect(Collectors.toMap(Map.Entry::getKey, Map.Entry::getValue, (prev, next) -> next, HashMap::new));

Answer 2

似乎Collectors.toMap没有获取stream.collect示例中的类型参数,只返回a Map<Object,Object>.

作为一种变通方法,您可以自己创建结果映射,并在最后一个流步骤中将过滤后的条目添加到结果映射中:

Map<Set<Integer>, Double> result = new HashMap<>();
container.entrySet()
    .stream()
    .filter(entry -> entry.getKey().size() == size)
    .forEach(entry -> result.put(entry.getKey(), entry.getValue()));