第7章

基礎(chǔ)案例案例介紹圖像繪制案例介紹第7章

今有中小板股票002001～002020共20個(gè)，以及2016年5月1日—2016年8月1日的交易數(shù)據(jù)StkcdTrddtClsprc0020012016-05-0320.990020012016-05-0420.420020012016-05-0520.490020012016-05-0618.70020012016-05-0918.670020012016-05-1019.07………………其中字段依次表示股票代碼、交易日期、收盤價(jià)。任務(wù)如下：（1）計(jì)算每只股票10日移動(dòng)平均收盤價(jià)格數(shù)據(jù)。（2）繪制每只股票收盤價(jià)走勢圖和移動(dòng)平均收盤價(jià)走勢圖，并用子圖的形式表示出來。案例介紹第7章

說明：每個(gè)figure按照2×2劃分，即每個(gè)figure繪制4只股票的收盤價(jià)走勢圖和移動(dòng)平均價(jià)格走勢圖，20只股票則需要5個(gè)figure，使用循環(huán)的方式實(shí)現(xiàn)。提示：可以考慮使用Python中的取余運(yùn)算來實(shí)現(xiàn)，取余符號為%，例如：0%4=01%4=12%4=23%4=34%4=05%4=16%4=27%4=38%4=09%4=1……凡是取余為0的時(shí)候都需要?jiǎng)?chuàng)建一個(gè)figure，在這個(gè)figure中繪制4個(gè)子圖。圖形繪制第7章

importpandasaspdimportnumpyasnpimportmatplotlib.pyplotasplttrd=pd.read_excel('trd.xlsx')#獲取股票代碼，并轉(zhuǎn)化為列表的形式c=trd['Stkcd'].value_counts()code=list(c.index)#動(dòng)態(tài)計(jì)算需要q個(gè)figure,#其中每個(gè)figure繪制4個(gè)子圖,每個(gè)子圖代表一個(gè)股票#初始值設(shè)置q=0q=0#循環(huán)對每一個(gè)股票繪制其圖形foriinrange(20):

#第i個(gè)股票的收盤價(jià)記為p,并計(jì)算其移動(dòng)平均價(jià)

#構(gòu)造繪圖的橫軸和縱軸坐標(biāo)值

p=trd.loc[trd['Stkcd'].values==code[i],'Clsprc']

avg_p=p.rolling(10).mean()

x1=np.arange(0,len(p))

y1=p.values

y2=avg_p[9:]

x2=np.arange(9,len(p))

#如果i與4整除,代表需要重新建一個(gè)figure（每個(gè)figure有4個(gè)子圖）

ifi%4==0:

q=q+1

plt.figure(q)

plt.figure(figsize=(8,6))

plt.subplot(2,2,i%4+1)

plt.tight_layout()#用于設(shè)置圖像外部邊緣自動(dòng)調(diào)整

plt.plot(