map 함수

python의 map() 함수는 지정된 iterable 및 map 객체의 모든 요소에 함수를 적용하는 데 사용된다.

map은 iterator이기 때문에 그 요소에 대해서 iterate할 수 있다.

 

map 함수는 다음과 같이 사용한다.

map(function, iterable, ...)

반복할 iterable에 대해서 각각 function들을 적용한다. 그리고 그 결과(map의 return 값)는 iterator이다.

 

 

Example 1. A와 B를 입력받아서 더한 값을 출력

A, B = map(int, input().split())
print(A+B)

c.f. input().split()

여러 input을 한 번에 받을 때 사용한다. 이 때 string 형태로 변수에 저장된다.

a, b, c = input().split()

여러 input을 int형으로 받고 싶을 때는 ex1의 답과 같이 map 함수를 사용하면 된다.

 

 

Example 2. list를 사용해서 소문자를 대문자로 변환

# Return double of n 
def addition(n): 
    return n + n 
  
# We double all numbers using map() 
numbers = (1, 2, 3, 4) 
result = map(addition, numbers) 
print(list(result))

output: [2, 4, 6, 8]

map object -> list

list(map)으로 map object -> list 변환 가능하다.

 

 

---

참고: https://newbedev.com/python-map-python-documentation-code-example