[Python] filter 함수

filter는 입력 받은 iterable object의 각 요소를 함수로 수행후 True인 것만 묶어서 반환합니다.

# 함수와 iterable 객체를 입력 받으며 반환 값은 boolean 값이어야 합니다.
filter(함수, <iterable object>)

양수 값만 반환하는 예제

numbers = [2, 1, -1, -2]

함수로 구현

def positive(numbers):
  result = []
  for number in numbers:
    if number > 0:
      result.append(number)
  return result

ret = positive(numbers)
print(ret)
# [2, 1]

filter로 구현

def positive(number):
  return number > 0

ret = list(filter(positive, numbers))
print(ret)
# [2, 1]

lambda와 filter로 구현

ret = list(filter(lambda x: x > 0, numbers))
print(ret)
# [2, 1]

References

https://wayhome25.github.io/cs/2017/04/03/cs-03/