[Python] 람다, 컴프리헨션

🔶 람다 표현식이란?

컴공 전공하면서 react 언어를 배우며 학습했던 람다.

코드를 간결하게 표현하는 방법 중 하나다.

람다를 표현한 식을 람다 표현식 이라고 부르고,

람다 표현식으로 쓴 함수는 람다 함수 혹은 익명 함수 라고 한다.

 

 

 

Q. 왜 익명함수 일까?

함수에 이름이 없어서.

 

 

 

표현 방식

(lambda 변수 : 리턴식 )

🔥 한 줄 요약

함수 이름이 없다. -> 따라서 함수를 변수에 할당하여 재호출하여야 한다. 
lambda 바로 뒤에 매개변수 작성한다. :(콜론)으로 구분하여 이후 리턴값 작성한다.
람다 표현식 안에 새 변수 생성 x
반환값 부분은 식 한 줄로 표현할 수 있어야 ok

 

 

 

---

람다 표현식을 인수로 사용하는 함수 3가지

• map(): 리스트의 요소를 지정된 함수로 처리해주는 함수
• filter(): 걸러주는, 필터링해주는 함수
• reduce(): 누적 합 계산해주는 함수

함수를 인수로 쓰는 코드를 작성할 때 효율적이다.

 

 

 

---

🔶 map()

알고리즘 코딩 문제에 종종 나오는 map() 함수는 리스트나 튜플 등에 특별한 처리를 할 때 사용하는 함수

list or tuple 형태로 반환

 

 

표현 방식

map( 함수, 적용할 값(튜플 or 리스트) )

 

 

 

파이썬3부터는 list() or tuple() 함수로 묶어주어야 원하는 결과를 리턴받을 수 있다.

list(map(function, list))
tuple(map(function, list))

 

 

 

🔶 map() + 일반 함수

# 일반 함수
def second(x):
    results = x * 2
    return results

# 리스트 
two = [i for i in range(10)]

# map() + 일반함수
list(map(second, two))

[0, 2, 4, 6, 8, 10, 12, 14, 16, 18]

 

 

🔶 map() + 람다 함수

# 리스트
two = [i for i in range(10)]

# map() + 람다 함수
list(map(lambda x : x * 2, two))

[0, 2, 4, 6, 8, 10, 12, 14, 16, 18]

 

 

 

---

🔶 filter()

리스트에 들어있는 원소들을 함수에 적용시켜서 결과가 참인 값들로 구성된 새로운 리스트를 만들어준다.

 

 

 

표현 방식

filter( 함수, 리스트 )

 

 

 

🔶 filter() + 일반 함수

# 일반 함수
def odd(x):
    if x % 2 == 1:
        return x

# 리스트
two = [i for i in range(10)]

# list(filter() + 일반 함수)
list(filter(odd, two))

[1, 3, 5, 7, 9]

 

 

 

🔶 filter() + 람다 함수

# list(filter() + 람다 함수)
list(filter(lambda x : x % 2, two))

 

 

 

관련 문제)

# Q. 다음 리스트를 출력하는 코드를 filter() 함수와 람다 표현식을 이용해 구현해보세요!
# [4, 5, 6]

quiz_list = [i for i in range(10)]
list(filter( lambda x: x>=4 and x<=6 , quiz_list ))

 

여기서 조건식에  x>=4 and x<=6  이렇게 작성하였지만

4 <= x <= 6 이렇게 적어도 된다.

파이썬에서는 체인비교 연산이 가능해서다.

 

 

 

---

🔶 reduce()

이터러블 객체의 각 요소를 지정된 함수로 처리한 뒤, 이전 결과에 누적해서 반환하는 함수이다.

 

 

 

표현 방식

reduce( 함수, 이터러블 객체 )

 

 

 

reduce()는 파이썬 3부터 내장 함수가 아니다.

따라서 functools 모듈에서 reduce()를 가져와야 한다.

🔥 한 줄 요약

import를 해야 한다는 말이다. 해당 코드는 다음과 같다.

from functools import reduce

 

 

 

🔶 reduce() + 일반 함수

# 모듈 import
from functools import reduce

# 일반 함수
def f(x, y):
    return x * y

# 리스트
a = [1, 2, 3, 4, 5]

# reduce(일반 함수, 이터러블 객체)
reduce(f, a)

120

 

 

 

🔶 reduce() + 람다 함수

# 리스트
a = [1, 2, 3, 4, 5]

# reduce(람다 함수, 이터러블 객체)
reduce(lambda x, y : x * y, a)

120

 

 

 

예시 문제) 

# Q. 리스트 컴프리헨션과 람다 표현식을 이용해 1부터 100까지의 합을 구하는 코드를 구현해보세요!
from functools import reduce

a = [i for i in range(1, 101)]
# print(a)

b = reduce(lambda x, y: x + y, a)
print(b)

누적이기 때문에 함수에는 인자를 2개 필요로 한다. 이를 이진함수 라고 부른다.

이렇게 lambda 함수를 사용하면 코드를 간결하게 작성할 수 있게 된다.

여기서, 컴프리헨션 개념이 나와서 아래에서 간략히 설명하고자 한다.

 

 

 

---

컴프리헨션

파이썬 문법 중 하나로, 여러 줄의 코드를 한 줄의 코드로 줄일 수 있어 직관적이다.

 

 

 

리스트 컴프리헨션

🔶 반복문과 컴프리헨션

# 반복문
list_for = []
for i in range(0, 11, 2):
    list_for.append(i)

print(list_for)

위의 코드를 컴프리헨션을 사용하면

 

# 리스트 컴프리헨션
[x for x in range(0, 11, 2)]

이렇게 줄일 수 있다.

빈 리스트를 선언하지 않아도, 가장 바깥에 대괄호[ ] 만 써주면 리스트를 생성하게 된다.

반복문을 대괄호 안에 쓴다.

반복문 앞에 변수만 쓰면 된다.

 

 

 

🔶반복문 + 조건문 컴프리헨션

 

 

형태

[ x for x in range(1, 11) if x % 2 == 0 ]

 

반복문 뒤에 조건문을 써준다.

 

 

 

🔶 중복 표현과 컴프리헨션

# for문 중복
sweetie = ['아이스크림', '크림빵', '젤리']
savory = ['감자칩', '팝콘', '뿌셔뿌셔'] 

[[x, y] for x in sweetie for y in savory]

# if문 중복
[x for x in range(1, 30) if x % 2 == 0 if x % 3 == 0]

 

컴프리헨션 안에 for문과 if문을 중복하여 사용할 수 있다.

 

 

 

---

셋 컴프리헨션 & 딕셔너리 컴프리헨션

리스트 컴프리헨션은 다른 자료형으로도 확장될 수 있다.

 

 

🔶 셋 컴프리헨션

# set 컴프리헨션을 만드는 방법
# 대괄호 대신에 중괄호 사용
{x for x in range(0, 11, 2)}

{0, 2, 4, 6, 8, 10}

 

 

 

 

🔶 딕셔너리 컴프리헨션

# {나폴리 선수 번호:이름} 딕셔너리
napoli_num = [3, 77, 9, 99]
napoli_player = ['minjae', '크바라츠헬리아', 'Osimhen', 'Anguissa'] 
{k:v for k, v in zip(napoli_num, napoli_player)}

# 딕셔너리 컴프리헨션 with if문
{k:v for k, v in zip(napoli_num, napoli_player) if k < 50}

zip()함수를 활용해서 key와 value를 지정하면 딕셔너리를 만들 수 있다.

if문도 함께 쓸 수 있다.