[Python知识库]第3关：文件读取与字符串处理 - 可视化显示股票数据

import?matplotlib?as?mpl

mpl.use('Agg')

import?matplotlib.pyplot?as?plt

?

import?datetime

?

def?Draw():

????appl?=?"step3/AAPL.csv"?#苹果

????google?=?"step3/GOOG.csv"?#谷歌

????ms?=?"step3/MSFT.csv"?#微软

????#在此绘制折线图

????#???请在此添加实现代码???#

????#?**********?Begin?*********#

????plt.xticks(rotation?=45)

????appdate,appopens?=?Read(open(appl))

????plt.plot(appdate,appopens,color?=?'red',linewidth?=?1.0,label="appl")

????gogdate,gogopens?=?Read(open(google))

????plt.plot(gogdate,gogopens,color?=?'green',linewidth?=?1.0,label="goog")

????msdate,msopens?=?Read(open(ms))

????plt.plot(msdate,msopens,color?=?'blue',linewidth?=?1.0,label="ms")

????plt.legend(["Apple","Google","Microsoft"])

????plt.ylabel("Open")

????plt.savefig("step3/output/data.png")

#如果有必要，可以增加别的函数协助完成任务，可在此添加实现代码?

#?**********?Begin?*********#

def?Read(file):

????dates=[]

????opens=[]

????file.readline()

????for?line?in?file.readlines():

????????i1?=?line.index(",",0,len(line))

????????dt?=?datetime.datetime.strptime(line[0:i1],"%Y-%m-%d").date()

????????dates.append(dt)

????????i2?=?line.index(",",i1+1,len(line))

????????opens.append(float(line[i1+1:i2]))

????file.close()

????return?dates,opens

????

#?**********?End?**********#