按日期的最大值过滤日期的TreeMap

时间:2015-11-14 05:15:51

标签: java java-8 java-stream treemap

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

2 个答案:

答案 0 :(得分:3)

可能的解决方案如下:

  • 在地图
  • 中的所有条目上创建一个Stream
  • 将该流收集到一个新映射中,其中键对应于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)按年份分组。分类器从LocalDate返回YearMonth个对象。 downstream收集器用于将具有相同年月的所有值收集到单个值中:在这种情况下,我们使用Collectors.maxBy(comparator)根据比较器比较每个条目来选择最大值{ {1}}键(comparingByKey)。由于此收集器返回LocalDate(如果Stream为空),我们将其包装到Collectors.collectingAndThen(downstream, finisher)中,其中finisher只返回可选值。因此,在此步骤结束时,我们有一个Optional

最后,我们保留此中间地图的values以将每个条目收集到新地图中,我们在其中明确创建Map<YearMonth, Map.Entry<LocalDate, Double>>。由于我们知道这里没有重复项,因此合并函数只会抛出TreeMap

示例输入/输出:

IllegalStateException

- &GT;

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

答案 1 :(得分:1)

我假设您正在使用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, ...}