利用python—matplotlib绘制股票K线图
成品展示：
用Python绘制股票K线图并不复杂，不过得做好一些准备工作例如相关库的安装，库的引入、日期格式的调整and表格转换
文章目录
利用python—matplotlib绘制股票K线图安装库导入库数据获取时间格式调整And格式转换k线图与均线图的绘制安装库
#Windows命令行窗口中输入并执行命令pip install /matplotlib/mpl_finance/archive/master.zip
导入库
import tushare as ts import matplotlib.pyplot as pltimport mpl_finance as mpfimport seaborn as snssns.set()
数据获取
#获取股票数据df = ts.get_k_data('000002','-01-01','-04-10')df
时间格式调整And格式转换
结果
*由于通过tushare模块获取的数据中时间部分格式无法普配，需要使用时间格式调整函数将时间格式更改为我们能够使用的数据类型
#导入调整日期格式涉及的两个库from matplotlib.pylab import date2numimport datetime#将Tushare 库获取到的日期数据转换成Candlestick_ochi()函数可读取的格式def date_to_num(dates):num_time = []for date in dates:date_time = datetime.datetime.strptime(date,'%Y-%m-%d')num_date = date2num(date_time)num_time.append(num_date)return num_time#将Dateframe 转换成二维数组，并利用date_to_num()函数转换成日期格式df_arr = df.valuesdf_arr[:,0]= date_to_num(df_arr[:,0])#将二维数组中日期转换成数字格式
k线图与均线图的绘制
#k线图绘制fig,ax=plt.subplots(figsize=(15,6))mpf.candlestick_ochl(ax,df_arr,width=0.6,colorup='r',colordown='g',alpha=1.0)plt.grid(True)ax.xaxis_date()
结运行结果：
简单绘制后K线图便出来了，但与网站对比发现我们还缺少均线图
#添加均线图df['MA5'] = df['close'].rolling(5).mean()df['MA10'] = df['close'].rolling(10).mean()
#完整K线图代码：plt.rcParams['font.sans-serif']=['SimHei']#设置正常现实中文fig,ax=plt.subplots(figsize=(15,6))mpf.candlestick_ochl(ax,df_arr,width=0.6,colorup='r',colordown='g',alpha=1.0)#添加均线图plt.plot(df_arr[:,0],df['MA5'])#五日均线plt.plot(df_arr[:,0],df['MA10'])#十日均线plt.grid(True)plt.title('万科A')plt.xlabel('日期')plt.ylabel('价格')ax.xaxis_date()
