[Python] map() 함수를 사용하는 방법

 

안녕하세요. 혀코입니다.

이번 시간에는 Python의 map 함수를 사용하는 방법에 대해서 알아보겠습니다.

map 함수는 두개 또는 세개의 parameter가 있어야 합니다.

map(function, iterable data)

첫번째 파라미터로 연산을 수행할 함수가 있어야 하고, 두번째와 세번째(옵션) 파라미터로 연산에 필요한 리스트나 튜플 형태의 데이터가 있어야 합니다.

 

리스트의 값을 제곱하기

# 리스트 정의
numbers = [1, 2, 3, 4, 5]

# 숫자의 제곱을 구하는 함수 정의
def powerNum(n):
    return n * n

# map을 사용해서 바로 출력하면 메모리 주소가 출력된다.
print(map(powerNum, numbers)) # <map object at 0x0000017C71797C70>
# map 함수 연산의 결과를 출력하려면 list로 변환하여 출력해야 한다. 
print(list(map(powerNum, numbers)) # [1, 4, 9, 16, 25]

 

함수 정의 대신 lambda 함수 사용하기

숫자의 제곱을 구하는 함수를 따로 정의했는데 이것을 아래와 같이 lambda를 써서 한줄로 표현할 수 있습니다.

 

# 리스트 정의
numbers = [1, 2, 3, 4, 5]

print(list(map(lambda x: x * x, numbers))) # [1, 4, 9, 16, 25]

 

두개 이상의 리스트를 연산하기

두개 이상의 리스트를 연산할 경우에는 다음과 같이 사용할 수 있습니다.

 

# 리스트 정의
numbers1 = [1, 2, 3, 4, 5]
numbers2 = [2, 4, 6, 8, 10]

print(list(map(lambda x, y: x * y, numbers1, numbers2))) # [2, 8, 18, 32, 50]

 

이렇게 Python의 map 함수를 사용하는 방법에 대해서 알아봤습니다.

해당 정보가 유용하셨다면, 공감과 구독 부탁 드립니다.

감사합니다. :)