[Python] map에서 lambda 쓰기

lambda는 다양한 함수에서 쓸 수 있는데 sort에서만 쓰다가 map에서 쓰는 법도 기록하려고 한다!

map(함수, 입력)

map은 입력 리스트를 함수와 같게 바꿀 수 있으며 나는 보통 map(int, list)과 같은 형태로 입력값을 한 번에 int로 바꾸는 방법으로 주로 사용했다.

제곱

아래의 코드 함수 부분에 제곱식을 넣어 list의 값을 제곱수로 바꿀 수 있다.

def calc(x):
    return x*x

print(list(map(calc, range(1,6))))

[1, 4, 9, 16, 25]

---

Lambda

위와 같이 함수를 넣어도 되지만 일회용으로만 사용할 함수의 경우 lambda로 작성하여 함수부분에 넣을 수 있다.

print(list(map(lambda x: x*x, range(1,6))))

[1, 4, 9, 16, 25]

---

a = [[1, 2],
     [3, 4],
     [5, 6]]

print(list(map(lambda x: x[0]*x[1], a)))

[2, 12, 30]