我有一个如下所示的集合,目前存储为排序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, ...}