이제 표 4-1과 식을 바탕으로 베타가 다른 종목으로 구성된 포트폴리오 전체의 베타를 파이썬으로 구해보자.
# dot 함수를 사용하기 위해 numpy를 임포트한다
import numpy as np
# 네이버 금융에서 조회한 세 종목의 52주 베타를 beta52 변수에 리스트 형태로 저장한다
beta52 = [ 1.38, 0.47, 1.02 ]
# 투자금액을 investment 변수에 리스트 형태로 저장한다
investment = [ 5, 2, 3 ]
# 투자 비중을 구하기 위해 미리 투자금액의 합계를 구한다
sumOfInvestment = sum( investment )
# 인라인 for 루프를 사용해 각 종목 투자금액을 전체 투자금액 합계로 나눠(w/sumOfInvestment) 투자 비중을 weights 변수에 리스트로 저장한다
weights = [ w/sumOfInvestment for w in investment ]
# dot 함수를 사용해 두 개의 리스트 investment, weight의 곱의 합계를 구한다
beta = np.dot( beta52, weights )
print( 'The portfolio beta is {0:0.2f}'.format( beta ) )