# sqrt 함수를 사용하기 위해 math 모듈을 임포트한다
import math

# 경기 국면별 확률과 주식의 기대수익률
stock_a = [ 0.07, 0.04, 0.01 ]
stock_b = [ 0.13, 0.04, -0.05 ]
prob = [ 1/3, 1/3, 1/3 ]

# 주식 a와 b의 경기 국면에 따른 수익률 기댓값을 저장할 변수를 준비한다
ex_a = 0.0
ex_b = 0.0

# 주식 a와 b의 기댓값을 구한다. 기대수익률 리스트와 경기 국면별 확률 리스트를 곱한 합계를 구한다
# 기대수익률과 경기 국면별 확률 리스트를 zip( ) 함수로 묶어 반복한다(zip( ) 함수는 1.5.1절 참고). 그리고 그 결과 리스트의 값은 s와 p로 받고 둘을 곱한 합계를 계산한다

for s, p in zip(  stock_a, prob ):vex_a = 
    ex_a + s*p

# 위의 두 줄 for 루프는 다음과 같이 한 줄의 인라인 for 루프로 바꿀 수 있다
ex_a = sum( s*p for s, p in zip( stock_a, prob ) )

for s, p in zip( stock_b, prob ):
    ex_b = ex_b + s*p