Java groupingBy 和转换字符串

Question

我收到一个字符串列表，我试图用.groupingBy() 进行分区，这将给我一个Map<String, List<String>>，并转换最终将出现在地图中的列表中的字符串。

Examples I've found 使用 Collectors.mapping，这是我想使用的，但所有示例都使用方法引用，例如 .mapping(SomeClass::getSomething, ...，因为我正在处理字符串，所以无法完成。

我以为我可以使用 lambda 作为映射函数，但似乎无法正确使用语法，最终出现“类型不匹配：无法转换...”错误。

这说明了我正在尝试做的类型——

import java.util.*;
import java.util.stream.Collectors;

class Main
{
  public static void main(String[] args)
  {
        List<String> inputs = Arrays.asList("D8", "S1", "S5", "D2", "D15", "S9");
        System.out.println(inputs);

        Map<String, List<String>>
        outputs = inputs.stream()
              .collect(
                  Collectors.groupingBy(s -> s.startsWith("D") ? "desperate" : "serious")
                  //,Collectors.mapping(n -> Integer.valueOf(n.replaceAll("[DS]", "")), Collectors.toList())
              )
              ;

        System.out.println(outputs);
  }
}

这将创建地图，按照我的预期对输入数据进行分组，并生成输出
{desperate=[D8, D2, D15], serious=[S1, S5, S9]}

但我想要什么，正如您可能从注释掉的行中猜到的那样
Collectors.mapping(n -> Integer.valueOf(n.replaceAll("[DS]", "")), Collectors.toList())
是一个Map<String, List<Integer>>，包含通过删除前缀从字符串中提取的值。

我的实际用例可能会将一个字符串转换为另一个字符串，或者从更复杂的字符串中提取整数，但这说明了我正在尝试做的事情。

在 groupingBy 时如何使用 lambda 函数进行映射？
从我读过的例子中，我也不清楚Collectors.mapping 中第二个参数toList() 的使用。

我用这段代码创建了一个repl。

Answer 1

Collectors.groupingBy() 的版本采用另一个收集器获取值，例如：

@Test
public void streamsTest() {
    List<String> inputs = Arrays.asList("D8", "S1", "S5", "D2", "D15", "S9");
    System.out.println(inputs);

    Map<String, List<Integer>> outputs = inputs.stream()
            .collect(
                    Collectors.groupingBy(
                            s -> s.startsWith("D") ? "desperate" : "serious",
                            Collectors.mapping(
                                    s -> Integer.valueOf(s.substring(1)),
                                    Collectors.toList())));

    System.out.println(outputs);
}

这会产生：

[D8, S1, S5, D2, D15, S9]
{desperate=[8, 2, 15], serious=[1, 5, 9]}

Answer 2

这类似于Erik's answer。不同之处在于，转变可以发生在分组之外，并且可以独立发展。

流、地图、组

1. 流式传输元素
2. 使用map 转换每个元素（这可以委托给单独的函数）
3. 定义收集策略
4. 基于收集策略的分组
5. 根据需要收集元素

import java.util.AbstractMap.SimpleEntry;
import java.util.Arrays;
import java.util.List;
import java.util.Map;
import java.util.function.Function;
import java.util.stream.Collectors;

class TransformAndGroupBy {

  public static void main(String[] args) {
    List<String> inputs = Arrays.asList("D8", "S1", "S5", "D2", "D15", "S9");

    System.out.println(inputs);

    Function<String, SimpleEntry<String, Integer>> transform = s -> s.startsWith("D")
      ? new SimpleEntry<>("desparate", Integer.parseInt(s.substring(1)))
      : new SimpleEntry<>("serious", Integer.parseInt(s.substring(1)));

    Map<String, List<Integer>> outputs = inputs.stream() // convert to stream
      .map(transform) // do the necessary transformation
      .collect(Collectors.groupingBy(e -> e.getKey(), Collectors.mapping(e -> e.getValue(), Collectors.toList()))); // group and collect as per the transformation
    System.out.println(outputs);
  }
}