使用 Java 流基于键过滤地图列表

Question

有List>，如下：

[
 {"A": "ABC","B": "ABCD","C": "0.945"},
 {"A": "ABC","B": "AB","C": "0.871"},
 {"A": "XYZ","B": "XYZA","C": "0.765"},
 {"A": "ABC","B": "ABD","C": "0.945"},
 {"A": "XYZ","B": "XYZB","C": "0.563"}
]

问题

Filter above List to get a Map where key as 'A' and highest value of 'C' for that 'A'.

{"ABC": "0.945", "XYZ": "0.765"}

初始代码

Map<String, String> m0 = new HashMap<>();
m0.put("A", "ABC");
m0.put("B", "ABCD");
m0.put("C", "0.945");

Map<String, String> m1 = new HashMap<>();
m1.put("A", "ABC");
m1.put("B", "AB");
m1.put("C", "0.871");

Map<String, String> m2 = new HashMap<>();
m2.put("A", "XYZ");
m2.put("B", "XYZA");
m2.put("C", "0.765");

Map<String, String> m3 = new HashMap<>();
m3.put("A", "ABC");
m3.put("B", "ABD");
m3.put("C", "0.945");

Map<String, String> m4 = new HashMap<>();
m4.put("A", "XYZ");
m4.put("B", "XYZB");
m4.put("C", "0.563");

List<Map<String,String>> res = Arrays.asList(m0,m1,m2,m3,m4);

我能得到：

[{A=ABC, B=ABCD, C=0.945}, {A=XYZ, B=XYZA, C=0.765}]

使用以下代码：

List<Map<String, String>> result = res.stream().collect(Collectors.groupingBy(map->map.get("A")))
                .entrySet().stream().map(m -> m.getValue().stream()
                        .max(Comparator.comparingDouble(s -> Double.parseDouble(s.get("C")))).get())
                .collect(Collectors.toList());

Answer 1

当您开始时丢弃不需要的条目并仅保留 A 和 C 的值作为 key/value 对以供使用，这样会更容易。

Map<String, Double> result = res.stream()
        .map(m -> Map.entry(m.get("A"), 
                            Double.parseDouble(m.get("C"))))
        .collect(Collectors.groupingBy(Entry::getKey,
                Collectors.mapping(Entry::getValue, 
                                   Collectors.maxBy(Double::compare))))
        .entrySet().stream()
        .collect(Collectors.toMap(Entry::getKey, e -> e.getValue().get()));

根据需要将 {ABC=0.945, XYZ=0.765} 生成为 result；但使用两个流来摆脱值的 Optional<Double> 类型。

使用 Java 流基于键过滤地图列表

Filter List of Map Based on Key Using Java Streams

java

java-8

java-stream