Matplotlib簡單例子
阿新 • • 發佈:2018-12-20
例子中包括畫了兩條曲線,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()
效果圖如下: