[Python] 내장함수 map 구현

이번에는 앞의 filter와 사용법이 비슷한 map에 대해서 알아보겠다.

 

map은 iterable의 모든 element에 function을 적용해서 반환한다.

 

사용 예시를 보자

 

test = [1, 7, 5, 2, 9, 11]

print(list(map(lambda x: x * 2, test)))
#[2, 14, 10, 4, 18, 22]

이렇게 test의 모든 원소 x에 대하여 x*2가 반환된다.

 

 

그럼 구현해보겠다

해당 iterable을 돌면서 각 원소에 function을 적용해주면 된다.

def my_map(function, iterable):
    res=[]
    for i in iterable:
      res.append(function(i))
    return res

 

아주 간단하게 구현된다.

 

그럼 테스트를 해보겠다.

 

assert list(my_map(lambda x: x * 2, test)) == list(map(lambda x: x * 2, test))

잘 구현된 것을 확인할 수 있다.