알면 유용한 파이썬 문법 (2) 리스트 컴프리헨션, lambda

리스트 컴프리헨션

💡 리스트 컴프리헨션 :  파이썬에서 리스트를 간결하게 생성하는 방법 

# 기본적인 구조
[표현식 for 항목 in iterable if 조건문]

• 표현식 :  각 항목에 대한 계산이나 변환
• 항목 :  반복되는 값
• iterable : 반복 가능한 객체

예시 1) 1부터 10까지의 숫자를 제곱한 리스트 생성

squares = [x**2 for x in range(1, 11)]
print(squares)  

## 출력: [1, 4, 9, 16, 25, 36, 49, 64, 81, 100]

• `range(1,11)`: 1부터 10까지 자연수를
• `for x `: x 로 받아서
• `x**2`: 제곱

예시 2) 리스트 컴프리헨션을 중첩하여 2차원 리스트 생성

matrix = [[i for i in range(1, 4)] for j in range(3)]
print(matrix)  
## 출력: [[1, 2, 3], [1, 2, 3], [1, 2, 3]]

• `range(1,4)`: 1부터 3까지 자연수를
• `i for i`: i로 받아서 i로 출력한 것을
• `range(3)`: 3번 반복하기

예시 3) if , else 조건문

[출력값 if 조건 else 조건 만족 x시 출력값 for i in 반복할 값]

→ if 조건문은 선택적으로 사용될 수 있으며, 조건이 참일 때만 해당 항목을 결과 리스트에 추가

Lambda

💡 Lambda :  익명 함수로, 이름 없이 정의되는 간단한 함수
주로 한 줄로 표현되며, 일반적인 함수 정의와는 달리 def 키워드를 사용하지 않고 lambda 키워드를 사용하여 정의

lambda 매개변수 : 계산식

예시 1) 제곱 함수

square = lambda x: x ** 2
print(square(4))  
## 출력: 16

→ 매개변수 x에 대하여 제곱하기(`x**2`)

 

예시 2) lambda & filter

💡 filter

filter(조건 함수, 반복 가능한 데이터)

numbers = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
even_numbers = list(filter(lambda x: x % 2 == 0, numbers))
print(even_numbers)  
## 출력: [2, 4, 6, 8, 10]

 

• 조건함수 자리에 `lambda x: x%2 == 0`이 들어감 (조건 : x가 짝수일 때)
• 반복 가능한 데이터 = numbers
  → numbers에서 x가 짝수인 경우를 even_numbers라는 리스트로 반환하기

예시 3)  lambda & map

💡 map

map(함수, 반복 가능한 데이터)

numbers = [1, 2, 3, 4, 5]
squared_numbers = list(map(lambda x: x ** 2, numbers))
print(squared_numbers)  
## 출력: [1, 4, 9, 16, 25]

• 함수 자리에 `lambda x: x**2`가 들어감 (x를 제곱해주는 함수)
• 반복 가능한 데이터 = numbers
  → numbers에 있는 모든 값들에 대해 제곱해 이를 squared_numbers라는 리스트로 반환하기