通过每个月的最大值过滤日期的TreeMap

Pat*_*ick 5 java treemap java-8 java-stream

我有一个类似下面的集合,当前存储为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
Run Code Online (Sandbox Code Playgroud)

Tun*_*aki 3

一个可能的解决方案如下:

  • 在 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)
        );
Run Code Online (Sandbox Code Playgroud)

在此代码中,首先使用 来按年月对地图进行分组Collectors.groupingBy(classifier, downstream)YearMonth分类器从 中返回一个对象LocalDate。收集downstream器用于将具有相同年月的所有值收集为单个值:在这种情况下,我们因此使用根据比较器比较每个条目键( )Collectors.maxBy(comparator)来选择最大值。由于此收集器返回 an (如果 Stream 为空),我们将其包装到 a 中,其中完成器仅返回可选值。在此步骤结束时,我们将得到一个.LocalDatecomparingByKeyOptionalCollectors.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
Run Code Online (Sandbox Code Playgroud)

->

2010-01-02=2108.74
2010-02-02=2217.92
2010-03-02=2327.57
2010-04-01=2427.57
Run Code Online (Sandbox Code Playgroud)