백테스트 시 거래기록 만들기

파이썬을 이용한 백테스트를 도전하고 있다. 

여러가지 방법이 있겠지만, 내가 할수 있는 방법을 기록하고자 한다.

 

2개의 리스트를 만든다.

하나는 for 반복문 이전에 만들고 하나는 for 반복문 내부에 만든다. 

book = []
for i in range(len(df)):
    day_list=[]

 

반복문 끝에 원하는 데이터를 내부리스트 (이 경우, day_list)에 담고, 

이 리스트를 외부리스트 (이 경우, book)에 append로 담는다. 

    day_list = [df.index[i],stock_cnt,total_asset,stock_own,own_day]    
    book.append(day_list)

 

이후 반복문 종료후 book 리스트를 데이터프레임으로 바꿔준다. 

이때 칼럼명을 지정한다.

book_df = pd.DataFrame(book, columns=['date','stock_cnt','total_asset','stock_own','own_day'])

결과값이 잘 나온다. 

만족스럽다.