java - 按每个月的最大值过滤日期的 TreeMap

Question

我有一个像下面这样的集合，目前存储为 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

Answer 1

可能的解决方案如下:

• 在 Map 中的所有条目上创建一个 Stream
• 将该 Stream 收集到一个新的 Map 中，其中键对应于 Map 的年月部分，值是当前条目。在重复的情况下，将只保留与日期有关的最大元素
• 根据该中间 Map 的值再次创建一个新 Stream
• 最后将其收集到 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) 按年月对 map 进行分组.分类器返回 YearMonth 来自 LocalDate 的对象. downstream收集器用于将具有相同年月的所有值收集到一个值中:在这种情况下，我们因此使用 Collectors.maxBy(comparator) 根据比较器比较每个条目选择最大值LocalDate键( comparingByKey )。由于此收集器返回 Optional (以防 Stream 为空)，我们将其包装成 Collectors.collectingAndThen(downstream, finisher) 终结者只返回可选值。因此，在这一步结束时，我们有一个 Map<YearMonth, Map.Entry<LocalDate, Double>>。 .

最后，我们保留 values 将这个中间映射的每个条目收集到一个新映射中，我们在其中显式创建一个 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