SeouliteLab

[Python/파이썬] reduce() 사용 방법 및 예제 본문

프로그래밍

[Python/파이썬] reduce() 사용 방법 및 예제

Seoulite Lab 2024. 3. 4. 08:47

파이썬의 reduce() 함수는 리스트와 같은 반복 가능한(iterable) 객체에 대해 주어진 함수를 왼쪽부터 오른쪽으로 적용하여 하나의 결과값을 반환합니다. 이 함수를 사용하면 여러 개의 값을 하나로 줄이는 데 유용합니다. 아래 예제를 통해 reduce() 함수의 사용법과 예제를 살펴보겠습니다.

1. reduce() 함수 기본 사용법

먼저, reduce() 함수를 사용하려면 functools 모듈을 import해야 합니다. 이 함수는 두 개의 인자를 받습니다. 첫 번째 인자는 적용할 함수이고, 두 번째 인자는 반복 가능한 객체입니다.

from functools import reduce

numbers = [1, 2, 3, 4, 5]
sum_all = reduce(lambda x, y: x + y, numbers)
print(sum_all)  # 출력: 15

위 코드에서는 리스트의 모든 요소를 더하는 예제입니다. lambda 함수를 사용하여 각 요소를 누적하여 합계를 구합니다.

2. 초기값 설정하기

두 번째 인자로 초기값을 설정할 수도 있습니다. 초기값을 설정하면 반복 가능한 객체가 비어있을 때 초기값을 반환합니다.

from functools import reduce

numbers = [1, 2, 3, 4, 5]
sum_all = reduce(lambda x, y: x + y, numbers, 10)
print(sum_all)  # 출력: 25

위 코드에서는 초기값을 10으로 설정하여 리스트의 모든 요소를 더할 때 빈 리스트일 경우에도 초기값이 반환되도록 합니다.