Java 流：将列表分组到地图中

java java-8 java-stream

2022-09-01 05:54:53

我如何使用Java Streams执行以下操作？

假设我有以下类：

class Foo {
    Bar b;
}

class Bar {
    String id;
    String date;
}

我有一个，我想把它转换为.即：首先按 Foo.b.id 分组，然后按 Foo.b.date 分组。List<Foo>Map <Foo.b.id, Map<Foo.b.date, Foo>

我正在为以下2步方法而苦苦挣扎，但第二个方法甚至没有编译：

Map<String, List<Foo>> groupById =
        myList
                .stream()
                .collect(
                        Collectors.groupingBy(
                                foo -> foo.getBar().getId()
                        )
                );

Map<String, Map<String, Foo>> output = groupById.entrySet()
        .stream()
        .map(
                entry -> entry.getKey(),
                entry -> entry.getValue()
                        .stream()
                        .collect(
                                Collectors.groupingBy(
                                        bar -> bar.getDate()
                                )
                        )
        );

提前致谢。

答案 1

您可以一次性对数据进行分组，假设只有不同的：Foo

Map<String, Map<String, Foo>> map = list.stream()
        .collect(Collectors.groupingBy(f -> f.b.id, 
                 Collectors.toMap(f -> f.b.date, Function.identity())));

使用静态导入保存一些字符：

Map<String, Map<String, Foo>> map = list.stream()
        .collect(groupingBy(f -> f.b.id, toMap(f -> f.b.date, identity())));

答案 2

假设对是不同的。如果是这样，在第二步中，您不需要分组，只需收集到键的位置，值本身就是：(b.id, b.date)Mapfoo.b.datefoo

Map<String, Map<String, Foo>> map = 
       myList.stream()
             .collect(Collectors.groupingBy(f -> f.b.id))    // map {Foo.b.id -> List<Foo>}
             .entrySet().stream()
             .collect(Collectors.toMap(e -> e.getKey(),                 // id
                                       e -> e.getValue().stream()       // stream of foos
                                             .collect(Collectors.toMap(f -> f.b.date, 
                                                                       f -> f))));

或者更简单：

Map<String, Map<String, Foo>> map = 
       myList.stream()
             .collect(Collectors.groupingBy(f -> f.b.id, 
                                            Collectors.toMap(f -> f.b.date, 
                                                             f -> f)));