본문 바로가기
분석하는 마케터

Python - filter 함수

by TREBOR 2019. 11. 4.

리스트에서 특정 조건을 만족하는 항목만 골라서 출력할 때, filter 함수를 활용할 수 있다.

먼저 0보다 큰 항목인지를 검토하는 함수를 정의해준다.

def pnum(data):
	return data>0

함수에 1이라는 인수를 대입하면 True라고 출력되지만,

[1, 2, 0, 5]라는 리스트를 대입하면 오류가 발생한다.

pnum(1)   #결과값: True
pnum      # TypeError : '>' not supported between instances of 'list' and 'int'

filter 함수를 활용하면 해결할 수 있다.

필터함수는 filter(함수명, 인수) 형태로 활용 예는 아래.

list(filter(pnum, [4,6,-5,-7,0]))  

# 결과: [4, 6]

댓글