[python 기초] map, filter 사용 + lambda

<예제 출처> 

https://wikidocs.net/22803

2) map, filter

 

 

사용법이 익숙치 않아 항상 찾아보는 map, filter함수를 확실히 정리해 놓는다. 

---

map(적용시킬 함수, 적용할 요소들) 

: 반복가능한 iterable 객체를 받아서 각 요소에 함수를 적용해주는 함수 

 

#1. for문 사용

def add_1(n): 
    return n + 1

target = [1, 2, 3, 4, 5]
result = []

for value in target: 
    result.append(add_1(value))
    
print(result)

[2, 3, 4, 5, 6]

 

#2. map함수 사용

# map 함수 사용 
def add_1(n): 
    return n + 1

target = [1, 2, 3, 4, 5]

result = map(add_1, target)

print(result)  # 출력결과: iterator -> next함수를 이용하여 확인 가능 
print(list(result))  # list 타입으로 형변환 하여 확인 가능

<map object at 0x000001B7457EF3C8>

[2, 3, 4, 5, 6]

 

#3. map함수 + lambda 사용

# map + lambda: add_1 과 같은 함수가 재사용 목적이 없다면 lambda 함수 사용
target = [1, 2, 3, 4, 5]

result = map(lambda x: x + 1, target)

print(list(result))

[2, 3, 4, 5, 6]

 

# 추가 예제: 모든 요소들을 str 타입으로 변경 
target = [1, 2, 3, 4, 5]
list(map(str, target))

['1', '2', '3', '4', '5']

---

filter(적용시킬 함수, 적용할 요소들) 

: 특정 조건으로 걸러서 걸러진 요소들로 iterator 객체를 만들어서 리턴 

#1. for문 사용

target = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
result = []

def is_even(n):
    return True if n % 2 == 0 else False 

for value in target: 
    if is_even(value): 
        result.append(value)
        
print(result)

 

#2. filter 함수 사용

[2, 4, 6, 8, 10]

# filter 함수 사용 
target = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]

def is_even(n): 
    return True if n % 2 == 0 else False

result = filter(is_even, target)

print(list(result))

[2, 4, 6, 8, 10]

 

#3. filter 함수 + lambda 사용

# filter + lambda
target = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
result = filter(lambda x: x%2==0, target)

print(list(result))

[2, 4, 6, 8, 10]

 

 

응용: Map + Filter 예제

## target리스트의 모든 요소들에 1을 더하고 홀수만 return
target = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
list(filter(lambda x: x%2!=0, map(lambda x: x+1, target)))

[3, 5, 7, 9, 11]