使用Java 8流分组通过在地图列表列表上吗?

罗伯特·科安尼(RóbertKohányi)

从以下输入

[
   [{group=a, value=cat}, {group=b, value=dog}],
   [{group=a, value=cow}, {group=b, value=bat}]
]

我如何获得以下输出

{
   a=[{value=cat}, {value=cow}],
   b=[{value=dog}, {value=bat}]
}

使用Java 8流?

我有以下标准解决方案

Map<String, String > map1 = new LinkedHashMap<>();
map1.put("group", "a");
map1.put("value", "cat");
Map<String, String > map2 = new LinkedHashMap<>();
map2.put("group", "b");
map2.put("value", "dog");
Map<String, String > map3 = new LinkedHashMap<>();
map3.put("group", "a");
map3.put("value", "cow");
Map<String, String > map4 = new LinkedHashMap<>();
map4.put("group", "b");
map4.put("value", "bat");

List<Map<String, String>> list1 = Arrays.asList(map1, map2);
List<Map<String, String>> list2 = Arrays.asList(map3, map4);

List<List<Map<String, String>>> input = Arrays.asList(list1, list2);

Map<String, List<Map<String, String>>> output = new LinkedHashMap<>();
for (List<Map<String, String>> list : input) {
   for (Map<String, String> map : list) {
       String group = map.get("group");
       if (!output.containsKey(group)) {
           output.put(group, new ArrayList<>());
       }
       List<Map<String, String>> values = output.get(group);
       values.add(map);
   }
}

我看到有一种Collectors.groupingBy方法,但是我不知道如何使用它。

Map<String, List<Map<String, String>>> output = input.stream()
  .<am I missing out some steps here?>
  .collect(Collectors.groupingBy(<what goes here?>))
塔吉尔·瓦列夫

要生成相同的输出,应flatMap在分组之前将列表的列表展平为单个列表

output = input.stream().flatMap(List::stream)
              .collect(Collectors.groupingBy(map -> map.get("group")));

此代码生成与您发布的命令性代码相同的输出:

{
 a=[{group=a, value=cat}, {group=a, value=cow}], 
 b=[{group=b, value=dog}, {group=b, value=bat}]
}

但是请注意,它与所需的输出不同。为了获得所需的输出,您可能需要指定下游收集器:

output = input.stream()
              .flatMap(List::stream)
              .collect(Collectors.groupingBy(map -> map.get("group"),
                 Collectors.mapping(
                     map -> Collections.singletonMap("value", map.get("value")), 
                      Collectors.toList())));

现在的结果是

{
  a=[{value=cat}, {value=cow}], 
  b=[{value=dog}, {value=bat}]
}

本文收集自互联网,转载请注明来源。

如有侵权,请联系 [email protected] 删除。

编辑于
0

我来说两句

0 条评论
登录 后参与评论

相关文章