본문 바로가기

Python/AI 수학 with Python

[Python] 총합과 총곱 계산

sum() 함수

$$y = \sum_{k=1}^5 a_k = a_1 + a_2 + a_3 + a_4 + a_5 $$

다음과 같은 수식을 sum() 함수를 통해 나타내보자.

[In]

import numpy as np
a = np.array([1,2,3,4,5]) # a1부터 a5까지 표현
y = np.sum(a) # 총합

print(y)

[Out]

15

prod() 함수

$$y = \prod_{k=1}^5 a_k = a_1 \times a_2 \times a_3 \times a_4 \times a_5 $$

다음과 같은 수식을 prod() 함수를 통해 나타내보자.

[In]

import numpy as np
a = np.array([1,3,2,5,4]) # a1부터 a5까지 표현
y = np.prod(a)  # 총곱

print(y)

[Out]

120