使用Java Streams转换嵌套列表

时间:2019-01-20 15:50:21

标签: java java-8 functional-programming java-stream

是否可以使用Stream s获得相同的结果(如我目前正在使用以下嵌套的for循环)?

List<String> people = Arrays.asList("John", "Adam");
List<String> dogs = Arrays.asList("alex", "rex");

List<List<String>> list = new ArrayList<List<String>>();
list.add(people);
list.add(dogs);

List<List<String>> list2 = new ArrayList<List<String>>();
for (int i = 0; i < list.size(); i++) {
    list2.add(new ArrayList<>());
    for (int j = 0; j < list.get(i).size(); j++) {
        list2.get(i).add(list.get(i).get(j).toUpperCase());
    }
}
System.out.println(list2);

我想得到这样的答复:

[[JOHN, ADAM], [ALEX, REX]]

使用以下Stream

list.stream().flatMap(l -> l.stream()).map(String::toUpperCase).collect(Collectors.toList());

我只能得到这样的东西:

[JOHN, ADAM, ALEX, REX]

1 个答案:

答案 0 :(得分:6)

flatMap对您无济于事,因为您不希望使用固定输出。您应该将每个内部List分别转换为大写List,然后将所有这些List收集到最终的嵌套输出List中。

List<List<String>> list2 = list.stream()
        .map(l -> l.stream().map(String::toUpperCase).collect(Collectors.toList()))
        .collect(Collectors.toList());

输出:

[[JOHN, ADAM], [ALEX, REX]]