画股票图

画股票图

import datetime
import matplotlib.pyplot as plt
from matplotlib import style
import pandas as pd
import pandas_datareader.data as web

if __name__ == '__main__':
    start = datetime.datetime(2018, 1, 1)
    end = datetime.date.today()
    stock = web.DataReader('002045.sz', 'yahoo', start, end)
    stock['60ma'] = stock['Adj Close'].rolling(window=60, min_periods=1).mean()
    ax1 = plt.subplot2grid((6, 1), (0, 0), rowspan=5, colspan=1)
    ax2 = plt.subplot2grid((6, 1), (5, 0), rowspan=1, colspan=1, sharex=ax1)
    ax1.plot(stock.index, stock['Adj Close'])
    ax1.plot(stock.index, stock['60ma'])
    ax2.bar(stock.index, stock['Volume'])
    plt.show()