使用流计算值的百分比

Question

我试图弄清楚在使用stream 时如何计算列表中特定值的百分比。

我的对象getTag 可以取值G 或Y，我想计算G:s 在列表中的百分比。另一个对象是getDateTime，其形式为1946-01-12。

我有条件filter(weather -> !weather.getDateTime().isAfter(dateTo) && !weather.getDateTime().isBefore(dateFrom))，因为我只想要用户输入的两个日期之间的百分比。

所以每个 DateTime 值对应于G 或Y。我用Collectors.counting()计算了G和Y的频率，但是如何得到百分比呢？

Map<String, Long> percentage = weatherData.stream()
        .filter(weather -> !weather.getDateTime().isAfter(dateTo) && !weather.getDateTime().isBefore(dateFrom))
        .collect(Collectors.groupingBy(Weather::getTag, Collectors.counting()));

Answer 1

如果您将您关心的项目 ("G") 映射为 1，而将其他所有项目映射为 0，则平均值是流中“G”的百分比。

double pctG = list.stream()
    .mapToInt(obj -> obj.getTag().equals("G") ? 1 : 0)
    .summaryStatistics()
    .getAverage();

使用Java 13，您可以使用teeing()收集器按标签统计元素，和过滤后的元素总数，最后将组数除以总数： p>

Map<String, Double> fractions = weatherData.stream()
        .filter(...)
        .collect(
            Collectors.teeing(
                Collectors.groupingBy(Weather::getTag, Collectors.counting()),
                Collectors.counting(),
                YourClass::scale));

scale() 函数将每个组除以总数：

static <T> Map<T, Double> scale(Map<? extends T, Long> counts, long total) {
    return counts.entrySet().stream().
        .collect(Collectors.toMap(e -> e.getKey(), ((double) e.getValue()) / total));
}