[python] map()

map 함수

각 요소에 적용할 함수(연산)과 컬렉션을 전달받아 컬렉션의 모든 요소에 연산을 적용하는 함수

---

<예시 1>

def plus(n):
	return n + 5

#range(5)의 각 요소에 plus 함수를 적용시켜 리스트로 만든다 
print(list(map(plus, range(5))

<출력 결과>

[5, 6, 7, 8, 9]

 

---

 

이때 적용할 함수가 간단하다면 lambda 식을 활용할 수 있다. 

<lambda 함수를 활용한 예시>

print(list(map(lambda n: n + 5, range(5))))

<출력 결과>

[5, 6, 7, 8, 9]

 

---

<예시 2>

map을 사용하여 리스트의 모든 요소를 int형으로 변환할 수 있다.

a_list = [1.1, 2.2, 3.3, 4.4]
print(list(map(int, a_list)))

<출력 결과>

[1, 2, 3, 4]

 

---

<예시 3>

정수형 변수를 스페이스바로 입력받기

num_list = list(map(int, input().split()))
print(num_list)

<입력>

1 2 3

<출력 결과>

[1, 2, 3]