python生成均线图，仅供自己复习使用

# -*- coding: utf-8 -*-
from __future__ import unicode_literals
import numpy as np
import datetime as dt
import matplotlib.pyplot as mp
import matplotlib.dates as md

def dmy2wday(dmy):
    dmy = str(dmy, encoding='utf-8')
    date = dt.datetime.strptime(dmy, '%d-%m-%Y').date()
    ymd = date.strftime('%Y-%m-%d')
    return ymd

dates, closing_prices =\
    np.loadtxt('C:/Users/Administrator/Desktop/asd/DS/data/aapl.csv',
               delimiter=',', usecols=(1, 6), unpack=True,
               dtype=np.dtype('M8[D],f8'),
               converters={1: dmy2wday})

# 使用有效卷积获取均线上点的价格
ma5 = np.convolve(closing_prices, np.ones(5) / 5, 'valid')

mp.figure('Moving Average', facecolor='lightgray')
mp.title('Moving Average', fontsize=20)
mp.xlabel('Date', fontsize=14)
mp.ylabel('price', fontsize=14)
ax = mp.gca()
ax.xaxis.set_major_locator(
    md.WeekdayLocator(byweekday=md.MO))

ax.xaxis.set_major_formatter(
    md.DateFormatter('%d %b %Y'))
mp.tick_params(labelsize=10)
mp.grid(linestyle=':')

dates = dates.astype(md.datetime.datetime)
mp.plot(dates, closing_prices, c='lightgray', label='Closing Price')
mp.plot(dates[4:], ma5, c='orangered', label='Md5')
mp.legend()
mp.show()