파이썬(Python) 내장함수 filter를 이용하여 리스트에서 특정 조건을 만족하는 원소 추출하기

---

오늘은 파이썬(Python) 내장 함수인 filter의 사용법을 알아보려고 한다. filter는 기본적으로 리스트에서 특정 조건을 만족하는 원소들만 추출하고 싶을 때 사용하는 함수다.

---

   filter 사용법

filter 함수는 첫 번째 인자로 조건을 나타내는 함수를 넣어준다. 조건을 나타내는 함수란 리턴값이 True 또는 False여야 한다는 것이다. 그리고 두 번째 인자로 추출하고 싶은 리스트를 넣어준다.

---

filter( '조건 함수', '리스트 또는 Iterable 가능한 객체' )

---

 

---

   filter 예제

여기서는 문자열 원소를 갖는 리스트에서 특정 문자로 시작하는 원소를 filter를 이용하여 추출해보고자 한다.

 

예를들어 문자열 리스트에서 'A'로 시작하는 문자열만 뽑는다고 한다면 다음과 같이 해준다. filter는 iterator 객체를 리턴하므로 추출된 결과를 리스트로 뽑기 위해선 list로 감싸준다.

 

word_list = ['Apple', 'Banana', 'Strawberry', 'Airplane', 'Ada'] ## 추출할 리스트

## 조건 함수
def func(x):
    return x[0] == 'A'

print(list(filter(func, word_list))) ## filter는 iterator 객체이므로 list로 감싼다.

 

---

오늘 코딩을 하다가 filter라는 엄청난 보물을 발견했다. 자주 사용할 것 같다.

---