Java - Stream의 filter(), map(), flatMap() 사용 방법

Java Stream의 filter(), map(), flatMap() 사용 방법 및 예제를 소개합니다.

1. Stream.filter()

filter()는 인자로 함수를 받으며, 어떤 조건으로 Stream의 요소들을 필터링합니다. 여기서 인자로 전달되는 함수를 구현하여 필터링하는 조건을 설정할 수 있습니다.

아래 예제는 filter()를 사용하여 Stream의 요소들 중에 "c"로 시작하는 문자열들을 모두 필터링합니다.

List<String> list =
        Arrays.asList("a1", "a2", "b1", "b2", "c2", "c1", "c3");
Stream<String> stream = list.stream();
Stream<String> filtered = stream.filter(s -> s.startsWith("c"));
filtered.forEach(System.out::println);

Output:

c2
c1
c3

2. Stream.map()

map()은 인자로 함수를 받으며, Stream의 요소를 다른 형태로 변경합니다. 인자로 전달되는 함수를 구현하여 요소를 어떻게 변경할지 설정할 수 있습니다.

아래 예제는 Stream의 문자열 요소들을 대문자로 변경합니다.

List<String> list =
        Arrays.asList("a1", "a2", "b1", "b2", "c2", "c1", "c3");
Stream<String> stream = list.stream();
stream.map(s -> s.toUpperCase()).forEach(System.out::println);

Output:

A1
A2
B1
B2
C2
C1
C3

메소드 레퍼런스 String::toUpperCase를 사용하면, 위와 동일하게 동작하는 코드를 아래와 같이 더 간단히 구현할 수 있습니다.

List<String> list =
        Arrays.asList("a1", "a2", "b1", "b2", "c2", "c1", "c3");
Stream<String> stream = list.stream();
stream.map(String::toUpperCase).forEach(System.out::println);

3. Stream.flatMap()

flatMap()은 평탄화라고 부르기도 하며, 복잡한 자료 구조에 저장된 요소들을 단순한 자료구조로 변환합니다.

예를 들어, Stream의 요소가 2차 배열인데, 2차 배열의 요소들을 모두 꺼내서 1차 배열에 넣어서 리턴합니다.

아래 예제에서 결과만 보면, Stream<String[]>Stream<String>로 변환되었습니다. 즉, 기존에 Stream<String[]>는 문자열 배열을 요소로 전달하는데, 이 배열의 모든 요소들을 꺼내서 평탄화하고 Stream<String>으로 문자열 단위로 요소를 전달하는 Stream을 만들었습니다.

String[][] arrays = new String[][]{ {"a1", "a2"}, {"b1", "b2"}, {"c1", "c2", "c3"} };
Stream<String[]> stream = Arrays.stream(arrays);
Stream<String> stream = stream.flatMap(s -> Arrays.stream(s));
stream.forEach(System.out::println);

Ouput:

a1
a2
b1
b2
c1
c2
c3

4. flatMap(), filter(), map()을 사용하는 예제

위에서 설명한 flatMap(), filter(), map()을 함께 사용하는 예제입니다.

flatMap()으로 2차 배열을 1차 배열로 평탄화하고, filter()"a"로 시작하는 문자열을 제외하며, map()으로 문자열들을 대문자로 변경하였습니다.

String[][] arrays = new String[][]{ {"a1", "a2"}, {"b1", "b2"}, {"c1", "c2", "c3"} };
Stream<String[]> stream = Arrays.stream(arrays);
Stream<String> stream = stream.flatMap(s -> Arrays.stream(s));
stream.filter(s-> s.startsWith("a"))
        .map(String::toUpperCase).forEach(System.out::println);

Output:

A1
A2
Loading script...

Related Posts

codechachaCopyright ©2019 codechacha