用Python绘制MACD、KDJ、布林线技术指标图

背景介绍：

搭建程序化交易平台的过程需要研究技术指标。

环境：

OS：mac

PYTHON: 3.8

IDE: pycharm

 

步骤：

第一步：加载库

import pandas as pd
import pandas_datareader as web
from datetime import datetime, timedelta
import talib
import matplotlib.pyplot as plt
from mplfinance.original_flavor import candlestick_ohlc
import matplotlib.ticker as ticker

pandas: 处理股票数据

pandas_datareader：行情数据获取api，也可以用tushare或者从新浪等网站获取

datetime：获取日期

talib：生成技术指标

candlestick_ohlc: 绘制蜡烛图

matplotlib.ticker: 设置x轴刻度，如果不设置，matplotlib会将没有交易的日期也显示出来

 

第二步：获取数据

def stock(stock_code): #stock_code是股票代码，例子：上市 "600036.ss", 深市 "000001.sz"
    start_date = "2019-12-01"  #起始日期
    today = datetime.date(datetime.now())  #截止日期
    stock_info = web.get_data_yahoo(stock_code, start_date, today)  #获取行情数据，返回dataframe
    stock_info = stock_info.reset_index()  #默认index是日期，这里要重置一下，为后面绘图做准备
    stock_info = stock_info.astype({"Date": str})    #将Date列的类型设置为str，为绘图做准备
    return stock_info