[python] 누적 곱 구하기

코테 문제를 풀면서 리스트의 모든 원소의 누적합이나 누적곱을 구해야 하는 경우가 많다.

그래서 누적곱을 쉽게 구하는 방법을 정리하였다.

 

 

1. reduce 사용

 - 리스트 내의 요소를 순차적으로 누적 계산한 결과를 리턴해준다.

from functools import reduce


## 방법 1 - 함수 사용
def func(x,y):
	return x*y
    
reduce(func, [1,4,2,5])



## 방법 2 - 람다 사용
reduce(lambda x,y : x*y , [1,4,2,5])

 

 

2. prod 사용

- math 라이브러리의 prod를 사용하여 구한다.

from math import prod

print(prod([1,4,2]))

# 8