Question

我有以下类型的Map

public class MapUtils {

    private Map<String, Integer> queryCounts = new HashMap<>();

public void averageCounters(){

    int totalCounts = queryCounts.values().stream().reduce(0, Integer::sum);
    queryCounts = queryCounts.entrySet()
                             .stream()
                             .collect(Collectors.toMap(
                                 Map.Entry::getKey,
                                 (Map.Entry::getValue)/totalCounts
    ));
}

这不会编译并在此行(Map.Entry::getValue)/totalCounts中显示错误。我该如何解决？有没有更好的方法可以使用Java 8 API获得Map的平均值？

修改：这是一种更好的方法吗？

queryCounts.entrySet()
           .forEach(entry -> queryCounts.put(entry.getKey(),
                                   entry.getValue()/totalCounts));

Answer 1

如果您想进行就地修改，最好使用Map.replaceAll代替Stream API：

int totalCounts = queryCounts.values().stream()
                             .collect(Collectors.summingInt(Integer::intValue));
queryCounts.replaceAll((k, v) -> v/totalCounts);

但是在您的情况下，此解决方案存在问题，因为除法结果将舍入为int数字，因此您几乎总是会在结果中得到零。实际上你的代码中存在同样的问题。您可能希望将Map<String, Double>作为结果类型。所以你可能需要创建一个全新的Map：

Map<String, Double> averages = queryCounts.entrySet().stream()
                                          .collect(Collectors.toMap(Entry::getKey,
                                              e -> ((double)e.getValue())/totalCounts));

另一种方法是首先将queryCounts声明为Map<String, Double>。这样您就可以使用replaceAll：

double totalCounts = queryCounts.values().stream()
                             .collect(Collectors.summingDouble(Double::doubleValue));
queryCounts.replaceAll((k, v) -> v/totalCounts);

最后，还有一个替代方案，它是最有效的，但很脏。您的代码假定在调用queryCounts后不需要原始（非平均）averageCounters()。因此，您可以将queryCounts保留为Map<String, Integer>（这比计算到Map<String, Double>更有效），但随后更改Map值类型如下：

double totalCounts = queryCounts.values().stream()
                             .collect(Collectors.summingInt(Integer::intValue));
Map<String, Object> map = (Map<String, Object>)queryCounts;
map.replaceAll((k, v) -> ((Integer)v)/totalCounts);
Map<String, Double> averages = (Map<String, Double>)map;
queryCounts = null;

在Collectors.groupingBy实现中的JDK中，类似的技巧是performed。

HashMap使用java 8流API的平均计数？

1 个答案: