[주가분석기초] 파이썬으로 평균거래량 찾기

종목의 기간내 평균거래량을 계산하는 함수를 소개합니다.

import pandas as pd

def get_mean_volume(filename):
    """filename.csv에 저장된 정보에서 각 종목의 평균거래량을 불러오는 함수입니다.
    참고: 파일이 저장된 위치는 다음과 같습니다: data/filename.csv
    """
    df = pd.read_csv("data/{}.csv".format(filename))
    # 데이터를 읽어옵니다
    return df['Volume'].mean()
    # 데이터를 계산하고 값을 출력합니다
    # 뒤의 .mean() 부분이 평균값을 계산합니다.

def test_run():
    """위의 함수를 테스트해보는 함수를 만듭니다 """
    for filename in ['AAPL', 'IBM']:
        print "Mean Volume"
        print filename, get_mean_volume(filename)

if __name__=="__main__": 
    test_run()