Java - 특정 조건으로 리스트 필터링, 3가지 방법

리스트의 요소들 중에 특정 조건을 만족하지 않는 요소들은 리스트에서 필터링하는 방법을 소개합니다.

1. List.removeIf()를 이용한 방법

removeIf()는 인자로 람다식을 받으며, 리스트의 모든 요소에 대해서 조건을 만족하면 리스트에서 요소를 삭제합니다.

아래 예제에서 리스트는 양수, 음수들을 갖고 있고, removeIf()로 음수들을 제거하였습니다. 이 방법은 원본 리스트가 변경된다는 특징이 있습니다. 원본 리스트를 보존해야 한다면 이 방법을 사용할 수는 없습니다.

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

public class Example {

    public static void main(String[] args) {
        List<Integer> list = new ArrayList<>(Arrays.asList(-2, -1, 0, 1, 2));

        list.removeIf((n) -> n < 0);
        System.out.println(list);
    }
}

Output:

[0, 1, 2]

2. for문을 이용한 방법

아래와 같이 반복문을 이용하여 리스트의 모든 요소를 순회하면서, 조건에 일치하는 요소들만 새로운 리스트에 저장할 수 있습니다.

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

public class Example1 {

    public static void main(String[] args) {
        List<Integer> list = new ArrayList<>(Arrays.asList(-2, -1, 0, 1, 2));

        List<Integer> filtered = new ArrayList<>();
        for (Integer n : list) {
            if (n >= 0) {
                filtered.add(n);
            }
        }
        System.out.println(filtered);
    }
}

Output:

[0, 1, 2]

3. Stream을 이용한 방법

for문과 같은 방식으로, Stream을 사용하여 요소들을 필터링하여 새로운 리스트에 저장할 수 있습니다. 스트림에서 filter()collect()를 사용합니다.

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.stream.Collectors;

public class Example2 {

    public static void main(String[] args) {
        List<Integer> list = new ArrayList<>(Arrays.asList(-2, -1, 0, 1, 2));

        List<Integer> filtered = list.stream()
                .filter((n) -> n >= 0)
                .collect(Collectors.toList());
        System.out.println(filtered);
    }
}

Output:

[0, 1, 2]
Loading script...
codechachaCopyright ©2019 codechacha