"Pandas DataFrame" 파이썬으로 배우는 알고리즘 트레이딩 스터디 자료

# Pandas DataFrame (스터디자료3. 파이썬으로 배우는 알고리즘 트레이딩)

# pandas의 Series가 1차원 자료구조라면 DataFrame은 여러개의 열로 구성된 2차원 형태의 자료구조입니다.

# 일자   시가   저가   고가    종가  전일비 등락률 거래량
# 2.26 11,250 11,100 11,850 11,800 +600 4.6% 335,587
# 2.25 10,900 10,500 11,450 11,200 +300 2.6% 125,484

# 위와 같은 자료를 DataFrame을 이용해서 2차원 구조의 데이터를 쉽게 저장하고 조작할 수 있습니다.

# DataFrame 객체를 생성하는 가장 쉬운 방법은 파이썬의 딕셔너리를 사용하는 것입니다.
# 딕셔너리를 통해 각 칼럼에 대한 데이터를 저장한 후
# 딕셔너리를 DataFrame 클래스의 생성자 인자로 넘겨주면 DataFrame 객체가 생성됩니다.
# raw_data 딕셔너리는 'col0','col1','col2'라는 키 값을 가지며 각 키는 리스트 타입의 값을 가집니다.
# 이를 DataFrame의 생성자로 넘겨주면 DataFrame 객체가 생성되는데 이대 생성된 객체를 data라는 변수가 바인딩합니다.

from pandas import Series, DataFrame

raw_data = {'col0': [1, 2, 3, 4],
'col1': [10, 20, 30, 40],
'col2': [100, 200, 300, 400]}

data = DataFrame(raw_data)
print(data)
'''
col0 col1 col2
0 1 10 100
1 2 20 200
2 3 30 300
3 4 40 400
'''

# 주의해서 살펴볼 부분은 col0, col1, col2라는 세 개의 칼럼이 존재한다는 점입니다.
# 여기서 'col0', 'col1', 'col2'라는 문자열은 DataFrame의 각 칼럼을 인덱싱하는 데 사용됩니다.
# 로우 방향으로는 Series와 유사하게 정수값이 자동으로 인덱싱된 것을 확인할 수 있습니다.

# 다음은 시가, 고가, 저가, 종가를 DataFrame으로 저장하는 것입니다.

daeshin = {'open': [11650, 11100, 11200, 11100, 11000],
'high': [12100, 11800, 11200, 11100, 11150],
'low' : [11600, 11050, 10900, 10950, 10900],
'close': [11900, 11600, 11000, 11100, 11050]}

daeshin_day = DataFrame(daeshin)
print(daeshin_day)
'''
close high low open
0 11900 12100 11600 11650
1 11600 11800 11050 11100
2 11000 11200 10900 11200
3 11100 11100 10950 11100
4 11050 11150 10900 11000
'''

# 열의 배열을 정하고 싶으면 columns를 사용해서 지정할 수 있습니다.
daeshin_day = DataFrame(daeshin, columns=['open', 'high', 'low', 'close'])
print(daeshin_day)
'''
open high low close
0 11650 12100 11600 11900
1 11100 11800 11050 11600
2 11200 11200 10900 11000
3 11100 11100 10950 11100
4 11000 11150 10900 11050
'''

# DataFrame 역시 인덱스 값을 넣으면 날짜기준 인덱싱을 할 수 있습니다.
date = ['16.02.29','16.02.26','16.02.25','16.02.24','16.02.23']
daeshin_day = DataFrame(daeshin, columns=['open', 'high', 'low', 'close'], index=date)
print(daeshin_day)
'''
open high low close
16.02.29 11650 12100 11600 11900
16.02.26 11100 11800 11050 11600
16.02.25 11200 11200 10900 11000
16.02.24 11100 11100 10950 11100
16.02.23 11000 11150 10900 11050
'''

스터디 자료의 출처는

파이썬으로 배우는 알고리즘 트레이딩(조대표 외 1명)입니다.

아래 책을 터치하시면 구매페이지로 연결됩니다. 

스터디 진행하며 사용한 비영리 목적의 참고자료로

크레이티브 커먼즈에 따라 사용하며,

저작권자의 삭제 요청이 있을시 삭제하겠습니다.

공감과 댓글은 글쓴이에게 큰 힘이 됩니다. 

마음에 드셨으면 공감과 댓글부탁드립니다.