如何使用Java 8中的流将列表按键分组并拆分值

Question

我需要对ABC列表进行分组，同时我也想拆分值。

输入：

List<ABC> list = new ArrayList<>();

ABC A1 = new ABC();
A1.setName("OO");
A1.setVal("OO111,OO222,OO333"); // split this while grouping
list.add(A1);

ABC A2 = new ABC();
A2.setName("OO");
A2.setVal("OO1");
list.add(A2);

ABC A3 = new ABC();
A3.setName("YY");
A3.setVal("1333,11444");
list.add(A3);

输出：

Map<String, List<String>> postsPerType;
//Group by Name: ex: OO & YY. Split the Value by "," and add it to list
     Key: 00 Values: List: "OO111","OO222","OO333","OO1"
     Key: YY Values: List: "1333","11444"

请提供建议，谢谢。

- krish

---

基本上，您可以使用split方法在逗号处拆分字符串。例如：String originalString = "OO111,OO222,OO333"; List<String> splitList = Arrays.asList(originalString.split(",")); - Ezani

Answer 1

如果您使用Java 9或更高版本，您可以使用Collectors.flatMapping()作为groupingBy的下游：

list.stream().collect(Collectors.groupingBy(ABC::getName,
                    Collectors.flatMapping(a -> Arrays.stream(a.getVal().split(",")), 
                                           Collectors.toList())));

那应该会精确地产生您所期望的结果。

- ernest_k

Answer 2

我认为你可以将一个 reducing 收集器链接到 groupingBy 收集器，以获得所需的结果：

Map<String, List<String>> postsPerType =
    list.stream()
        .collect(Collectors.groupingBy(ABC::getName,
                                       Collectors.reducing(new ArrayList<String>(),
                                                           a -> Arrays.asList(a.getVal().split(",")),
                                                           (l1,l2)-> {
                                                             List<String> l = new ArrayList<>(); 
                                                             l.addAll(l1);
                                                             l.addAll (l2);
                                                             return l;
                                                           })));

这会生成以下的Map：

{YY=[1333, 11444], OO=[OO111, OO222, OO333, OO1]}

- Eran

Answer 3

另一种方法是使用groupingBy和collectingAndThen收集器。

list.stream()
            .collect(Collectors.groupingBy(
                    ABC::getName,
                    Collectors.collectingAndThen(
                            Collectors.mapping(ABC::getVal, toList()),
                            strings -> strings.stream()
                                     .flatMap(s -> Arrays.stream(s.split(",")))
                                     .collect(toList()))
            ));

- Hadi J

Answer 4

尝试一下。

record ABC(String name, String val) {
    public String getName() { return name; }
    public String getVal() { return val; }
}

public static void main(String[] args) {
    List<ABC> list = List.of(
        new ABC("00", "00111,00222,00333"),
        new ABC("00", "001"),
        new ABC("YY", "1333,11444"));
    Map<String, List<String>> postsPerType = new HashMap<>();
    for (ABC e : list)
        postsPerType.computeIfAbsent(e.getName(), k -> new ArrayList<>())
            .addAll(Arrays.asList(e.getVal().split(",")));
    System.out.println(postsPerType);
}

输出：

{YY=[1333, 11444], 00=[00111, 00222, 00333, 001]}

- user4910279