Map이란?
map은 스트림 내부의 요소 하나하나에 접근해서 제가 파라미터로 넣어준 함수를 실행한 뒤 최종연산에서 지정한 형식으로 반환해주는 메서드 입니다.
List<Person> people = Arrays.asList(
new Person("personA", 24),
new Person("personB", 26),
new Person("personC", 28),
new Person("personD", 30)
);
위와 같은 리스트를 예시로 만들어보겠습니다.
List<String> nameList = people.stream()
.map(person -> person.getName())
.collect(Collectors.toList());
map 메서드 내부를 보면 람다 표현식이 인자로 들어간 것을 알 수 있습니다.
람다 표현식은
- people 리스트의 Person 객체 하나하나에 접근한다.
- getName 메서드를 통해 name값을 얻는다.
- name값들을 포함하는 list를 최종적으로 만들어서 반환한다.
라는 과정을 나타냅니다.
스트림을 쓰지 않을때는 for문을 사용해서 요소 하나하나에 접근해야 했지만, map이 반복문의 역할을 대신해주는 것입니다.
nameList = [personA, personB, personC, personD]
결과를 출력했을때 정상적으로 name값들로 만들어진 리스트가 반환된 것을 알 수 있습니다.
flatMap이란?
map에 대해서는 간략하게 알아봤는데, flatMap은 어떤 기능을 가지고 있을까요? flatMap은 스트림 평면화라는 이름으로도 불립니다. flatMap은 예시를 먼저 보는것이 이해하기 쉽기 때문에 아래의 코드를 한번 보겠습니다.
animal = ["cat","dog"]
위의 리스트에 스트림을 적용하여 cat과 dog의 철자 하나하나를 분리해서 새로운 리스트로 만드는 작업을 진행해보겠습니다.
원하는 결과 = [ "c", "a", "t", "d", "o", "g" ]
먼저 map을 사용해서 철자들을 분리해보겠습니다.
List<String[]> results = animals.stream().map(animal -> animal.split(""))
.collect(Collectors.toList());
분명 split 메서드를 이용하여 철자를 모두 분리했는데 반환값 타입을 보면 List<String[]>이 나온것을 알 수 있습니다.
[ [ "c", "a", "t" ] , [ "d", "o", "g" ] ]
즉, 위의 그림과 같이 animals 리스트 내부에서 각각의 요소별로 리스트을 중복으로 만들었다고 할 수 있습니다. 우리가 원하는 것은 리스트가 중복되는 것이 아니라 하나의 스트림에 철자 하나하나가 다 들어가는 것입니다. 이때 사용되는것이 flatMap, 다시 말해 중복된 스트림을 1차원으로 평면화 시키는 메서드입니다.
List<String> results = animals.stream().map(animal -> animal.split(""))
.flatMap(Arrays::stream)
.collect(Collectors.toList());
여기서 flatMap 내부의 Arrays::stream은 배열을 스트림으로 변환해주는 메서드 참조 표현입니다. flatMap을 사용하면 각각의 String 리스트를 스트림으로 만드는 것이 아니라, String 리스트를 감싸고 있는 더 큰 스트림의 컨텐츠, 즉 구성 요소로 만들어 줄 수 있습니다.
results = [c, a, t, d, o, g]
이처럼 flatMap을 사용하면 중복 구조로 되어있는 리스트를 하나의 스트림처럼 다룰 수 있습니다.
'JAVA > Java' 카테고리의 다른 글
| Stream.reduce() (1) | 2024.07.15 |
|---|---|
| LocalDate (LocalDate to Date, Date to LocalDate) (0) | 2023.08.24 |
| [RegExp] 전화번호, 휴대폰번호, 주민번호 정규식 (0) | 2021.09.10 |
