使用java 8 Stream按第一个值对字符串数组进行分组

Question

我有一个包含两个字符串的数组列表：

List<String[]> pets = {["Alice", "dog"], ["Bob", "horse"], ["Alice", "cat"], ["Bob", "cat"]}

我想把它们变成一个从字符串到字符串列表的映射，如下所示：

Map<String, List<String>>> pets = Map where
"Alice" maps to the List {"dog, "cat"}
"Bob" maps to the List {"horse", "cat"}

即我们取数组的第一个元素，并将第一个元素彼此相等的所有元素组合在一起。解决这个问题的必要方法是遍历元素，并逐步构建一个 Map。有没有办法用 java Streams 来做到这一点，即使用 Collector 或 reduce？

Answer 1

当然可以！

您可以按每个数组的第一个元素进行分组，这将是映射键，然后应用映射收集器来获取相应的值。

 Map<String, List<String>> resultSet = pets.stream()
                .collect(groupingBy(array -> array[0],
                        collectingAndThen(mapping(e -> Arrays.copyOfRange(e, 1, e.length),
                                toList()),
                                f -> f.stream().flatMap(Arrays::stream)
                                .collect(toList()))));

或：

 Map<String, List<String>> resultSet = pets.stream()
                .collect(groupingBy(array -> array[0],
                flatMapping(e -> Arrays.stream(Arrays.copyOfRange(e, 1, e.length)),
                                    toList())));

注意，此解决方案使用 flatMapping 收集器，该收集器仅在 JDK-9 中可用。