Python - filter() 사용 방법 및 예제

파이썬에서 filter() 함수는 리스트와 같은 iterable 객체의 요소들을 어떤 기준으로 필터링합니다.

예를 들어, 반복문으로 아래와 같이 리스트의 요소들 중에 25보다 큰 값만 필터링하도록 구현할 수 있는데, filter 함수를 이용하면 더 적은 코드로 구현할 수 있습니다.

numbers = [20, 10, 44, 12, 34, 40]
filtered = []

for num in numbers:
    if num >= 25:
        filtered.append(num)

print(filtered)

Output:

[44, 34, 40]

1. Syntax

filter()는 함수와 iterable 객체를 인자로 받습니다. iterable은 리스트와 같이 필터링할 요소들이 있는 객체이고, 함수는 어떤 기준으로 필터링할지에 대한 내용이 있습니다.

filter(function, iterable)

2. 리스트에 filter()를 사용하는 예제

다음과 같이 filter()에 함수와 리스트를 전달하여 25보다 큰 숫자만 필터링할 수 있습니다.

def myFunc(num):
    return num > 25

numbers = [20, 10, 44, 12, 34, 40]

filtered_iter = filter(myFunc, numbers)
filtered_list = list(filtered_iter)

print(filtered_list)

Output:

[44, 34, 40]

filter()는 iterator를 리턴합니다. 리스트로 변환하려면 list(iterator)처럼 list()로 감싸줘야합니다.

3. lambda와 filter()를 사용하는 예제

함수 대신에 lambda를 filter()의 인자로 전달할 수도 있습니다.

아래 코드는 위의 예제에서 함수 대신에 lambda를 사용하였습니다. lambda를 사용하는 것이 더 간단합니다.

numbers = [20, 10, 44, 12, 34, 40]

filtered_iter = filter(lambda num: num > 25, numbers)
filtered_list = list(filtered_iter)

print(filtered_list)

Output:

[44, 34, 40]

4. 이중 리스트에 filter()를 사용하는 예제

아래와 같이 이중 리스트에 filter()를 사용할 수도 있습니다. 아래 코드는 lambda에서 리스트의 정수 부분만 비교하여 1500보다 큰 값을 갖고 있는 리스트만 남기고 있습니다.

fruits = [
    ["apple", 1100],
    ["kiwi", 2100],
    ["banana", 1400],
    ["blueberry", 3300]
]

filtered_iter = filter(lambda fruit: fruit[1] > 1500, fruits)
filtered_list = list(filtered_iter)

print(filtered_list)

Output:

[['kiwi', 2100], ['blueberry', 3300]]
Loading script...
codechachaCopyright ©2019 codechacha