当前位置：首页 > article >正文

java stream流map和flatmap的区别

article 2024/11/13 1:22:31

map和flatmap都是用来转换操作。
map()操作后的流与原始流的元素个数一一对应，一对一地进行元素转换，适用于对每个元素进行简单的转换操作，例如将元素的属性提取出来或进行数值计算。
flatMap()操作是一对多的元素转换，对于每个输入元素，都可以生成多个输出元素。适用于将一个元素展开为多个元素的场景，例如将列表展开为单独的元素，或者将字符串拆分为单词。

示例:
1.map获取字符长度
假设有一个包含多个单词的字符串列表，获取每个单词的长度，并生成一个包含所有单词长度的列表。


List<String> words = Arrays.asList("Hello", "World", "Java", "Stream", "Example");
List<Integer> wordLengths = words.stream()
                                 .map( w->w.length() )
                                 .collect(Collectors.toList());
//输出[5, 5, 4, 6, 7]

在这个例子中，map()操作将每个字符串映射为它们的长度，生成了一个新的流。最后通过collect()方法将结果收集到列表中。结果列表为[5, 5, 4, 6, 7]。

2.flatmap割字符串成单词，再统计长度
flatMap()操作将每个句子拆分为单词，并生成一个包含所有单词的流。接着使用map()操作将每个单词的长度映射为新的流。最后通过collect()方法将结果收集到列表中。结果列表为[5, 5, 4, 6, 7]。


        List<String> sentences = Arrays.asList("Hello World", "Java Stream", "Example");
        List<Integer> wordLengths1 = sentences.stream()
                .flatMap(str -> Arrays.stream(str.split(" ")))
                .map( w-> w.length())
                .collect(Collectors.toList());
        System.out.println(wordLengths1);
                                     
//输出[5, 5, 4, 6, 7]

3.假设有一个包含多个人员的列表，每个人员都有多个技能，我们想获取所有人员的技能列表。


@Data
class Person {
    private String name;
    private List<String> skills;
    Person(){

    }
    Person( String name , List<String> skills){
        this.name = name ;
        this.skills = skills;
    }
}

flatMap()操作将每个人员的技能列表转换为一个流，并将多个技能流合并为一个扁平化的流。最后通过collect()方法将结果收集到列表中。结果列表为[“Java”, “Python”, “SQL”, “JavaScript”, “HTML”, “CSS”, “C++”, “C#”, “Ruby”]。

        List<Person> people = Arrays.asList(
                new Person("John", Arrays.asList("Java", "Python", "SQL")),
                new Person("Emily", Arrays.asList("JavaScript", "HTML", "CSS")),
                new Person("Michael", Arrays.asList("C++", "C#", "Ruby"))
        );

        List<String> allSkills = people.stream()
                .flatMap(person -> person.getSkills().stream())
                .collect(Collectors.toList());
        System.out.println(allSkills);
//输出[Java, Python, SQL, JavaScript, HTML, CSS, C++, C#, Ruby]