[python] map 함수 사용법

한 줄에 띄어쓰기로 구분된 문자열이나 숫자를 받을 때 map함수를 사용하는데,

map함수에 대해 알아보자

 

map(Function, iterable Data)

역할 : 반복 가능한 자료형의 모든 요소에 첫번째 매개 변수로 준 함수를 적용한 결과를 반환한 map객체 반환

• 첫번째 매개 변수 : 함수
• 두번째 매개 변수 : 반복 가능한 자료형

 

def add(n):
	return n + 1

arr = [1, 2, 3]

add_arr = map(add, arr)

print(add_arr)	# map 객체가 출력됨

# map 객체이기 때문에 list로 반환하여 출력
print(list(add_arr))	# [2, 3, 4]

 

map함수를 이용한 여러개의 정수 한번에 입력 받기

# map 함수 이용하지 않은 경우
n_list = list(input().split())
num1 = int(n_list[0])
num2 = int(n_list[1])
print(num1, num2)

# map 함수 이용한 경우
num1, num2 = map(int, input().split())
print(num1, num2)