按每个月的最大值筛选日期树状图



我有一个如下所示的集合,目前存储为排序TreeMap。请注意,每个月有多个条目。如何使用 Java 8 流按每个月中的最大值过滤它?

date=value
2010-01-01=2100.00, 
2010-01-02=2108.74, 
2010-02-01=2208.74, 
2010-02-02=2217.92, 
2010-03-01=2317.92, 
2010-03-02=2327.57, 
2010-04-01=2427.57, 
2010-04-02=2437.67, 
2010-05-01=2537.67, 
2010-05-02=2548.22, 
2010-06-01=2648.22, 
2010-06-02=2659.24, 
2010-07-01=2759.24, 
2010-07-02=2770.72, 
2010-08-01=2870.72, 
2010-08-02=2882.66, 
2010-09-01=2982.66, 
2010-09-02=2995.07, 
2010-10-01=3095.07, 
2010-10-02=3107.94, 
2010-11-01=3207.94, 
2010-11-02=3221.29

可能的解决方案如下:

  • 在映射中的所有条目上创建流
  • 将该流收集到新地图中,其中键对应于地图的年月部分,值是当前条目。如果重复,则仅保留与日期有关的最大元素
  • 在该中间映射的值上再次创建新的流
  • 最后收集成TreeMap.

假设初始 Map 的类型为 TreeMap<LocalDate, Double> ,这将是一个实现(此代码使用来自 Collectors 类的静态导入(:

TreeMap<LocalDate, Double> filtered =
        map.entrySet()
           .stream()
           .collect(groupingBy(
                e -> YearMonth.from(e.getKey()), 
                collectingAndThen(maxBy(Map.Entry.comparingByKey()), Optional::get))
           )
           .values()
           .stream()
           .collect(toMap(
                Map.Entry::getKey, 
                Map.Entry::getValue,
                (v1, v2) -> { throw new IllegalStateException(); },
                TreeMap::new)
        );

在此代码中,地图首先使用 Collectors.groupingBy(classifier, downstream) 按年月分组。分类器从LocalDate返回一个YearMonth对象。downstream收集器用于将所有具有相同年月的值收集到单个值中:在这种情况下,我们使用Collectors.maxBy(comparator)根据比较器选择最大值,比较每个条目LocalDate键(comparingByKey(。由于此收集器返回一个Optional(如果 Stream 为空(,因此我们将其包装到一个Collectors.collectingAndThen(downstream, finisher)中,其中完成器只返回可选值。因此,在此步骤结束时,我们有一个 Map<YearMonth, Map.Entry<LocalDate, Double>> .

最后,我们保留此中间映射的values,以将每个条目收集到新 Map 中,在其中我们显式创建一个TreeMap .由于我们知道这里没有重复项,因此合并函数只是抛出一个 IllegalStateException .

示例输入/输出:

2010-01-01=2100.00
2010-01-02=2108.74
2010-02-01=2208.74
2010-02-02=2217.92
2010-03-01=2317.92
2010-03-02=2327.57
2010-04-01=2427.57

->

2010-01-02=2108.74
2010-02-02=2217.92
2010-03-02=2327.57
2010-04-01=2427.57

我假设你为此使用LocalDate

TreeMap<LocalDate, Double> map = new TreeMap<>();
// ...

这是一个功能性解决方案

Map<Month, Optional<Double>> max =
map.entrySet()
   .stream()
   .collect(Collectors.groupingBy(
       e -> e.getKey().getMonth(),
       TreeMap::new,
       Collectors.mapping(
           e -> e.getValue(),
           Collectors.maxBy(Comparator.naturalOrder())
       )));
System.out.println(max);

哪些打印

{JANUARY=Optional[2108.74], FEBRUARY=Optional[2217.92], MARCH=Optional[2327.57], ...}

这是一个势在必行的解决方案(以防万一(

Map<Month, Double> max = new TreeMap<>();
for (Entry<LocalDate, Double> entry : map.entrySet()) {
    Month month = entry.getKey().getMonth();
    Double value = max.get(month);
    if (value == null || value < entry.getValue())
        max.put(month, entry.getValue());
}
System.out.println(max);

哪些打印

{JANUARY=2108.74, FEBRUARY=2217.92, MARCH=2327.57, ...}

最新更新