Java groupingBy 和转换字符串

如何解决Java groupingBy 和转换字符串

我收到了一个字符串列表，我试图用 .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);
  }
}

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

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

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

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

我使用此代码创建了一个 repl。

解决方法

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]}

,

这类似于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","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);
  }
}