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