pyplot (1) 썸네일형 리스트형 [python pandas] matplotlib.pyplot으로 주식 차트에 buy sell 표시하기 주식 데이터를 pyplot을 통해서 시각화 하던중, buy와 sell을 한 날짜를 표시해주고 싶었다. 그럼 아래 코드를 통해서 buy (초록색), sell(빨간색)을 차트 상의 종가에 표시해줄 수 있다. 전제조건은 df 의 column 안에 buy와 sell이 표시되어 있어야 한다. 아래 예제에서는 buy 값이 1인 경우 매수, sell 값이 1인 경우 매도를 뜻한다. import matplotlib.pyplot as plt # 그림 사이즈 설정 plt.figure(figsize=( 30,6)) # 종가 기준으로 차트를 그림 df['close'].plot(grid=True) # buy가 1 인경우 초록색으로 해당 종가에 표시 plt.scatter(df.loc[df.buy == 1].index, df.cl.. 이전 1 다음