# 포트폴리오 기대수익률 계산
# numpy 라이브러리를 np라는 이름으로 임포트한다
import numpy as np

# 자산의 개수를 정한다. 자산의 개수는 난수를 만들 때 사용할 것이다
numStocks = 3

# 세 가지 경기 국면별로 자산의 개수만큼 주식의 수익률을 난수로 생성한다
# [3 x numStocks] 배열을 만들고 만들어진 수익률 배열을 출력한다
returns = np.random.randn( 3, numStocks )
print( '1. 난수로 만드는 국면별 주식의 수익률: \n', returns )

# 세 가지 경기 국면별 확률을 만든다. 이것 역시 난수로 만든다
# 세 가지 국면의 확률의 전체 합이 1.0이 되도록 한다
# 난수를 세 개 만든다
prob = np.random.rand( 3 )

# 생성한 난수를 난수의 합계로 나눠 합이 1.0이 되도록 한다
prob = prob / prob.sum( )
print( '2. 경기 국면별 각 확률: \n', prob )