# -*- 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()