파이썬 데이터분석

파이썬(python) map함수

황원진 2023. 8. 3. 13:38

map:

리스트와 같이 여러 요소를 갖는 객체의 요소를 한번에 바꿔주는 함수이다.

 

말로만 하면 이해가 어려우니 예시와 사용법을 바로 보도록 하자

 

map()

map(function,sequence)

function은 함수를

sequence는 리스트와 같이 여러 요소를 갖는 객체를 넣어준다.

예시를 한번 보자

다음은 문자열 리스트에서 각 문자열의 길이를 구하는 예시이다.

def get_length(string):
    return len(string)

string_list = ["apple", "banana", "cherry", "date"]
length_list = list(map(get_length, string_list))
print(length_list)  


# 출력하면 [5, 6, 6, 4]가 나온다.

map의 첫번째 부분에 맨 처음 정의했던 get_length를 넣었고

map의 두번째 부분에 리스트인 string_list를 넣었다.

 

그 후 map함수의 반환값은 map객체이기 때문에 이를 list로 바꾸기 위해서 앞에 list를 붙였다.

 

이해가 되지 않는다면 쉬운 예시를 한번 더 들어보자.

다음은 리스트의 모든 요소를 제곱하여 새로운 리스트를 만드는 것이다.

def square(x):
    return x * x

my_list = [1, 2, 3, 4, 5]
squared_list = list(map(square, my_list))
print(squared_list)  # 출력: [1, 4, 9, 16, 25]

 

 

첫번째 예시와 크게 차이가 없다. 

 

 

이렇게 map()을 사용하면 반복문으로 처리해야 하는 과정들을 간단하게 처리할 수 있다.