python 画K线图,用python画星空,#数据源:http:


#数据源:http://pan.baidu.com/s/1hs5Wn0w

#要求:Matlibplot简单应用

#1.根据数据绘制出2017年6月~8月日线级别的价格走势K线。

#2.将MA5、MA10、MA20叠加至图中

import pandas as pdimport numpy as npimport matplotlib.pyplot as pltimport matplotlib.finance as mpfimport datetimefrom matplotlib.pylab import date2numplt.rcParams[‘font.sans-serif‘] = [‘SimHei‘]plt.rcParams[‘axes.unicode_minus‘] = False#读取数据并去掉多余数据data=pd.read_csv(‘000001.SZ.csv‘,encoding=‘gbk‘,index_col=0).iloc[:-2,:4]#将索引调整为datetime格式data.index=pd.to_datetime(data.index)#将分钟数据聚合为日数据data_open=data.loc[:,‘开盘价(元)‘].resample(‘D‘).first().dropna()data_high=data.loc[:,‘最高价(元)‘].resample(‘D‘).max().dropna()data_low=data.loc[:,‘最低价(元)‘].resample(‘D‘).min().dropna()data_close=data.loc[:,‘收盘价(元)‘].resample(‘D‘).last().dropna()#将开盘、收盘、最高、最低数据合并,注意数据顺序,与candlestick_ochl一致new_data=pd.concat([data_open,data_close,data_high,data_low],axis=1)#new_data=new_data.ix[‘2017-06‘:‘2017-08‘];print(new_data)#将日期索引调整到列new_data=new_data.reset_index()#将日期转换为num格式lst=[]for i in range(len(new_data)):    lst.append(int(date2num(new_data.iloc[i,0])))new_data[‘日期‘]=lstquotes=np.array(new_data)fig,ax=plt.subplots(figsize=(8,5))mpf.candlestick_ochl(ax,quotes,width=1,colorup=‘g‘,colordown=‘r‘)#分别画出5日、10日、20日均线图new_data.index=new_data[‘日期‘]pd.rolling_mean(new_data[‘收盘价(元)‘],5).plot()pd.rolling_mean(new_data[‘收盘价(元)‘],10).plot()pd.rolling_mean(new_data[‘收盘价(元)‘],20).plot()#将x轴设置为日期,调整x轴日期范围ax.xaxis_date()ax.set_xlim(datetime.datetime(2017,6,1),datetime.datetime(2017,8,31))plt.show()

python 画K线图

评论关闭