본문 바로가기

Python/AI 수학 with Python

[Python] 평균과 기댓값 구현

 

python에서 평균은 numpy의 average() 함수로 구현할 수 있다.

[In]

# 평균
import numpy as np

x = np.array([55, 45, 60, 40]) # 평균을 취할 데이터

print(np.sum(x)/4)
print(np.average(x))

[Out]

50.0
50.0

 

python에서 기댓값은 numpy의 sum() 함수로 구현할 수 있다.

 

예제 1

[In]

# 기댓값
import numpy as np

p = np.array([0.8, 0.15, 0.05]) # 확률
x = np.array([100, 500, 1000]) # 값

print(np.sum(p*x)) # 기댓값

[Out]

205.0

 

예제 2

[In]

# 기댓값
import numpy as np

p = np.array([0.75, 0.23, 0.02]) # 확률
x = np.array([100, 500, 10000]) # 값

print(np.sum(p*x)) # 기댓값

[Out]

390.0