Java 8 中按映射列表分组 8

2022-09-04 02:37:23

我有一个这样的列表:

List<Map<String, Long>>

有没有办法使用lambda将此列表转换为:

Map<String, List<Long>>

例:

Map<String, Long> m1 = new HashMap<>();
m1.put("A", 1);
m1.put("B", 100);

Map<String, Long> m2 = new HashMap<>();
m2.put("A", 10);
m2.put("B", 20);
m2.put("C", 100);

List<Map<String, Long>> beforeFormatting = new ArrayList<>();
beforeFormatting.add(m1);
beforeFormatting.add(m2);

格式化后:

Map<String, List<Long>> afterFormatting;

它看起来像这样:

A -> [1, 10]
B -> [100, 20]
C -> [100]

答案 1

您需要在每个 Map 的条目集中创建一个 .然后,可以使用分组 By(分类器,下游)收集器收集此流:分类器返回条目的键,下游收集器将条目映射到其值并将其收集到 .flatMapStream<Map.Entry<String, Long>>List

Map<String, List<Long>> map = 
     list.stream()
         .flatMap(m -> m.entrySet().stream())
         .collect(groupingBy(Map.Entry::getKey, mapping(Map.Entry::getValue, toList())));

此代码需要以下静态导入:

import static java.util.stream.Collectors.groupingBy;
import static java.util.stream.Collectors.mapping;
import static java.util.stream.Collectors.toList;

通过您的完整示例:

public static void main(String[] args) {
    Map<String, Long> m1 = new HashMap<>();
    m1.put("A", 1l);
    m1.put("B", 100l);

    Map<String, Long> m2 = new HashMap<>();
    m2.put("A", 10l);
    m2.put("B", 20l);
    m2.put("C", 100l);

    List<Map<String, Long>> beforeFormatting = new ArrayList<>();
    beforeFormatting.add(m1);
    beforeFormatting.add(m2);

    Map<String, List<Long>> afterFormatting =
        beforeFormatting.stream()
                        .flatMap(m -> m.entrySet().stream())
                        .collect(groupingBy(Map.Entry::getKey, mapping(Map.Entry::getValue, toList())));

    System.out.println(afterFormatting); // prints {A=[1, 10], B=[100, 20], C=[100]}
}

答案 2

推荐