본문 바로가기
  • 데이터야 놀자
데이터 분석 환경

파이썬 데이터프레임 Dataframe Pivot 결과 plotly로 차트 그리기

by 데이터랑 2023. 4. 9.
728x90

엑셀에서 피봇기능은 매우 편리해서 자주 사용한다.

그런데 파이썬 Pandas에서도 피봇기능이 가능한 것을 알았다.

 

Pandas 데이터프레임(Dataframe) 피봇기능(Pivot) 이용하여

코스피 콜옵션 종가 차트를 출력하는 방법을 간단히 정래해 보았다.

 

 

코스피 콜옵션 Dataframe 피봇 (Pivot) 결과 Plotly로 차트 그리기

 

 

Dataframe 피봇 (Pivot)
Dataframe 피봇 (Pivot)

 

  • sqlite3에서 데이터를 읽어와 데이터프레임을 생성하고
  • 그 데이터를 피봇으로 변환
  • 피봇데이터를 plotly를 활용해 그래프로 출력

 

sql_option = f"select date_tr, code_option, price_open, price_high, price_low, price_close from '{table_name}' \
                        where gb_option = '3'"

df = pd.read_sql(sql_option, conn)
df.sort_values(by=[df.columns[0]])


df['date_tr'] =  pd.to_datetime(df['date_tr'])

pvdf = pd.pivot_table(df,                # 피벗할 데이터프레임
                     index = 'date_tr',    # 행 위치에 들어갈 열      => 데이터프레임에서 인덱스 지정기능 겸함
                     columns = 'code_option',    # 열 위치에 들어갈 열
                     values = 'price_close',     # 데이터로 사용할 열
                     aggfunc = 'sum')   # 데이터 집계함수
print(pvdf)
pvdf.fillna(method='pad')           ###결측치 처리 : 앞의 행값
pvdf.interpolate(method='linear', limit_direction='forward')  ###결측치처리 보간법
plyo.iplot(pvdf.iplot(asFigure=True))

 

< 브라우저에 출력된 결과 >

Plotly 활용한 콜옵션 종가 그래프 그리기
Plotly 활용한 콜옵션 종가 그래프 그리기


 

 

728x90
728x90

댓글