python(vscode)/주식 차트그리기 #2/candle 차트/봉 차트 그리기/ 키움 api/키움증권

지난 글에서 차트를 그렸다면 이번에는 차트에 색상을 추가해보자.

 

*지난글*

https://gongdolgongdol.tistory.com/115 

python(vscode)/주식 차트그리기 #1/candle 차트/봉 차트 그리기/ 키움 api/키움증권

 

 

*지난 코드*

---

 

import mplfinance as mpf
import pandas as pd
from api.Kiwoom import *    
 
## 1. 일봉 데이터 확보
df = Kiwoom.get_price_data("005930") # 일봉데이터 가져와 df 에 넣기
print(df)                                                  # 데이터 프레임 출력
print(df.info())                                         # 데이터 정보 출력
 
## 2. 데이터 구조 변경
df['date'] = pd.to_datetime(df['date'])  # 문자열 데이터를 시계열 타입으로 변경한다
print(df.info())                                     # 데이터 정보 출력
print(df)                                              # 데이터 프레임 출력
df.set_index('date', inplace = True)    # 데이터 프레임내 인덱스 설정
print(df)                                              # 데이터 프레임 출력
 
## 3. 그래프 그리기
mpf.plot(data = df.loc["2022-10"], type='candle') # 22년 10월 데이터 프레임 그래프 그리기

---

 

이제 색상을 추가해보자.

*전체 코드*

---

 

import mplfinance as mpf
import pandas as pd
from api.Kiwoom import *  
## 1. 일봉 데이터 확보
df = Kiwoom.get_price_data("005930") # 일봉데이터 가져와 df 에 넣기
print(df)                            # 데이터 프레임 출력
print(df.info())                     # 데이터 정보 출력
 
## 2. 데이터 구조 변경
df['date'] = pd.to_datetime(df['date']) # 문자열 데이터를 시계열 타입으로 변경한다
print(df.info())                        # 데이터 정보 출력
print(df)
df.set_index('date', inplace = True)    # 데이터 프레임내 인덱스 설정
print(df)
 
## 추가. 색상 추가하기
mc = mpf.make_marketcolors(
    up="r",
    down="b"
)
 
s = mpf.make_mpf_style(
    base_mpf_style="starsandstripes",
    marketcolors=mc,
    gridaxis='both',   # horizontal, vertical, both
    y_on_right=True    # False는 y축을 왼쪽에 표시
)
 
## 3. 그래프 그리기
mpf.plot(data = df.loc["2022-10"], type='candle',style=s)

 

---

 -. 양봉은 빨간색, 음봉은 파란색이다. 

 

결과는 아래와 같다. 

 

 

 

거래량 , 이동평균 그래프를 다음글에서 확인해보자. 

 

https://gongdolgongdol.tistory.com/118

python(vscode)/주식 차트그리기 #3/거래량/이동평균/candle 차트