map()

1. map 함수

반복 가능한 객체의 요소를 지정된 함수로 처리하는 함수이다.

• map(function, iterable)
  • function: 각 요소에 적용할 함수
  • iterable: 함수를 적용할 데이터 집합

 

2. 리스트에 map 사용

a = [1.2, 2.5, 3.7, 4.6]
for i in range(len(a)):
    a[i] = int(a[i])

print(a) # [1, 2, 3, 4]

실수가 저장된 리스트 a가 있을 때, 리스트 a의 모든 요소를 정수를 변환하기 위해 for 반복문 사용했다.

 

번거로우니 map을 사용해 보자.

a = [1.2, 2.5, 3.7, 4.6]
a = list(map(int, a))

print(a) # [1, 2, 3, 4]

map에 int와 리스트를 넣으면, 리스트의 모든 요소를 int를 사용해서 변환한다.

그다음에 list를 사용해서 map의 결과를 다시 리스트로 만든다.

 

사실 map에는 리스트뿐만 아니라 모든 반복 가능한 객체를 넣을 수 있다.

a = list(map(str, range(10)))

print(a) # ['0', '1', '2', '3', '4', '5', '6', '7', '8', '9']

range로 0 ~ 9 사이의 숫자를 만들고, str을 이용해서 모두 문자열로 변환했다.

 

3. input().split()과 map

input().split()의 결과가 문자열 리스트라서 map 사용이 가능하다.

a = map(int, input().split()) # 입력: 10 20 
print(list(a)) # [10, 20]