파이썬 map 내장함수 사용하기

내 마음대로 파이썬 - zip 내장함수 사용하기

알고리즘 문제를 풀면서 자주 사용하는 파이썬 내장함수에 대해 살펴보고 있습니다.

이전에 zip 함수에 대해 정리한 바가 있는데요, 이번 정리에서는 파이썬의 내장함수인 map 함수에 대해 알아보겠습니다.

 

map(function, iterable)

map 함수는 첫 번째 인자로 함수를 전달 받습니다.

이때 전달 받은 함수를 두 번째 인자인 iterable 객체(리스트, 튜플 등)의 각 요소에 대해 적용합니다.

 

def double(x):
    return x * 2

print(type(map(double, [3, 6, 9])))  # <class 'map'>
print(type(list(map(double, [3, 6, 9]))))  # <class 'list'>

map 함수를 실행하면 리턴되는 객체는 map 타입입니다.

map 타입 객체는 리스트 또는 튜플로 형 변환이 가능합니다.

위 예제 코드는 map 함수에 리스트 [3, 6, 9]를 전달해서 리스트 내 각 요소에 2를 곱하는 작업을 수행합니다.

 

list(map(lambda x: x * 2, [10, 20, 30]))  # [20, 40, 60]

간단한 작업을 수행하는 함수 혹은 짧은 코드의 함수는 람다 함수로 대체할 수 있습니다.

위 예제 코드는 앞서 살펴본 함수를 람다 함수의 형태로 바꿔 map 함수에 인자로 전달합니다.