1. 程式人生 > >Matplotlib簡單例子

Matplotlib簡單例子

例子中包括畫了兩條曲線,X軸為日期,含有圖例。

import matplotlib.pyplot as plt
import tushare as ts

#通過tushare獲取資料
df = ts.get_hist_data('60001t6')
df1 = ts.get_hist_data('600012')    
    
y = df['close'].tolist()
#取得資料是從最近日期開始,故畫圖時需反向,注意:他是在原list裡反向
y.reverse()
#獲取的日期,這裡沒用上
x = df.index.tolist()

#下面是獲得第二條曲線的資料
y2 = df1['close'].tolist()
y2.reverse()
x2 = df.index.tolist()

#建立一個影象
fig, ax = plt.subplots()
#設定x軸的範圍
ticks = ax.set_xticks([0,len(x)/5,len(x)*2/5,len(x)*3/5,len(x)*4/5,len(x)])
#設定X軸顯示的標籤
labels = ax.set_xticklabels([x[len(x)-1], x[int(len(x)*4/5)],  x[int(len(x)*3/5)],  x[int(len(x)*2/5)], x[int(len(x)/5)], 0], rotation=30, fontsize='small')
#畫第一條線
plt.plot(y,label='600016')
#畫第二條線,‘r--'為線的形狀為---,顏色為紅色,顯示圖例標籤為600012
plt.plot(y2,'r--',label='600012')
#顯示圖例
ax.legend(loc='best')
plt.show()

效果圖如下: