Daily Python #4 : map 함수란?

map 함수는 list와 같은 iterable 객체 각 요소에 특정 함수를 적용시켜주고 그 결과값을 묶어서 돌려주는 파이썬 내장함수입니다. 이런 map을 어떻게 사용할 수 있을지 한번 알아보도록 하겠습니다.

 

 

이번 글에서 알아볼 내용

• map 함수란?
• map 함수 사용예시

 

map 함수란?

map 함수는 위에서도 언급했듯이 iterable 객체의 각 요소에 특정 함수를 적용시켜주는 내장함수입니다. 이 말을 이해하기 위해서는 일단 map 함수의 형태를 이해해야 합니다.

 

map(function, iterable) # 반환값은 list가 아니라 Iterator 입니다.

 

map 함수는 위에 보시다시피 function 과 iterable 을 인자값으로 받습니다. 이것을 실제 데이터로 치환해서 보시면 이해가 쉬워지실 것 입니다.

 

li = [1, 2, 3, 4]

# map의 반환값이 list가 아닌 Iterator 이기 때문에 형변환이 필요합니다.
result = list(map(lambda a : a + 1, li))


# 결과 [2, 3, 4, 5]

 

위와 같은 결과가 나오게 된 이유는 li 라는 list의 요소 하나하나에 1을 더하는 함수를 적용했기 때문입니다. 더 다양한 예시는 아래 map 함수 사용예시에서 알아보도록 하겠습니다.

 

map 함수 사용예시

map함수는 굉장히 유연하게 사용가능한 함수입니다. 따라서 다양한 상황에서 적용되는데요 그 중에서도 대표적인 것들을 짚어 설명해드리겠습니다.

 

1. input().split() 데이터의 int 형 변환

map 을 사용하는 가장 대표적인 예시입니다. input().split() 을 하면 각각의 요소가 char 형태로 있기 때문에 int 연산이 필요한 경우 형변환을 해주어야 합니다. 다음은 map 함수를 활용하여 int 형변환을 적용시켜 list 형태로 돌려주는 코드입니다.

 

li = list(map(int, input().split()))

# 만약 input이 "1 2 3 4 5 6" 이라면 결과값은 [1, 2, 3, 4, 5, 6]

 

해당 코드는 알고리즘을 Python으로 풀이하신다면 필수적으로 사용하실 겁니다.

 

2. 어떤 iterable 객체에 일괄적인 작업을 진행해야 할때

만약 어떤 list의 모든 요소를 전부 특정한 수식에 맞게 변경시켜야 한다면 어떻게 하시겠습니까? 첫 번째 방법은 for문을 돌려서 이 모든 것을 처리하는 방법입니다. 하지만 해당 방법은 파이썬 스럽지 않습니다. 왜냐하면 map을 사용하면 훨씬 짧고 단순한 코드를 이를 해결할 수 있기 때문입니다. 다음 코드는 약간 복잡한 계산을 list 각각의 요소에 적용시켜 보겠습니다.

 

#원의 둘레를 구하는 함수
def getRound(radius):
    	return int(radius * 2 * 3.14)
    
radius_list = [1, 2, 3, 4, 5]

round_list = list(map(getRound, radius_list))

# 결과값 [6, 12, 18, 25, 31]

 

꽤 복잡한 연산을 진행했습니다. 반지름을 받아와서 원의 둘레를 구한 후 int형으로 형변환하여 반환했습니다. 만약 이 작업을 for 문으로 작성했다면 위의 코드처럼 깔끔하게 나올 수 없었을 것 입니다.

 

물론 lambda를 이용해서 작성할 수도 있었지만 함수를 정의하고 사용할 수 있다는 것을 보여드리고 싶었습니다.

 

마무리하면서

이렇게 해서 파이썬 내장함수 map에 대한 포스트를 마무리하겠습니다. 앞으로 여러분들의 Iterable 객체를 다루어야 하는 상황이 오시면 map 함수를 효과적으로 사용하시길 바라겠습니다. 긴 글 읽어주셔서 감사합니다.