[Python] map() 사용하기

최근에 백준 문제를 파이썬으로 풀면서 map()함수를 자주 사용하게 되었다.

그래서 map()함수의 쓰임에 대해서 정리해보고자 한다.

 

 

map()의 매개변수는 다음과 같다. 

함수를 넣고, 그 함수를 적용할 리스트, 셋, 튜플 등을 넣어주면 된다.

map(function, iterable, ...)

 

 

리스트에 들어있는 숫자들을 문자로 바꿔주고 싶다면

아래 처럼 사용하면 된다.

str함수를 temp의 원소에 사용하겠다는 의미

그리고 list()해주어야 리스트의 형태로 만들어진다.

직접 만든 함수도 사용 가능하다.

temp = [1,2,3,4]
temp2 = list(map(str,temp))
temp2

 

 

람다 함수를 이용할 수도 있다!

temp = [1,2,3,4]
temp2 = list(map(lambda x : x**2, temp))
temp2

 

 

또한, 함수에 따라 여러 매개변수를 넣을 수도 있다.

temp = [1,2,3,4]
temp2 = [5,6,7,8]
temp3 = list(map(lambda x,y:x+y, temp, temp2))
temp3

 

 

백준 문제를 풀 때, 맨 처음에 input값을 받아와야한다.

그럴 때에도 map()을 사용하면된다 !

map(int, input().split())

 

 

 

참고 : https://www.programiz.com/python-programming/methods/built-in/map