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

# 분산을 저장할 변수와 투자 비중을 미리 준비한다
var_a = 0.0
var_b = 0.0
wgt_a = 0.5
wgt_b = 0.5

# 위 코드 네 줄은 변수에 값을 할당하는 코드인데, 다음과 같이 한 줄로 처리할 수 있다
var_a, var_b, wgt_a, wgt_b = 0.0, 0.0, 0.5, 0.5

# 리스트 stock_a와 prob에서 각각 데이터를 변수 s와 p로 받아 반복한다
# 확률과 편차 제곱을 곱해 var_a에 저장한다
for s, p in zip( stock_a, prob ):
    var_a = var_a + p*( s-ex_a )**2

var_a = sum( p*( s-ex_a )**2 for s, p in zip( stock_a, prob ) )