python绘制股票k线图
使用python绘制股票k线图
1. 需要安装的包
- tushare
- matplotlib
- mpl_finance
- datetime
使用Anaconda Prompt安装，安装语句’pip install 包的名字’
2. 获取数据
使用tushare包导入数据
import tushare as ts
df=ts.get_k_data('000002',start='2019-06-01',end='2019-09-30')
第一个参数’000002‘为股票代码，第二个参数start表示起始时间，第三个参数end表示结束日期。获取的数据df为DataFrame格式的二维表格。
3.调整日期格式
因绘制股票k线图使用的是mpl_finance库candlestick_ochl函数，该函数只能读取数组格式的数据，且其日期数据必须是特定的数字格式，因此，需要对原始数据中的日期数据进行格式的调整，调整代码如下：
import datetime
from matplotlib.pylab import date2num
#定义转换函数
def date_to_num(dates):
num_time=[]
for date in dates:
date_time=datetime.datetime.strptime(date,'%Y-%m-%d'))
num_date =date2num(date)
num_time.append(num_date)
return num_time
将df数据转换为二维数组，并调用date_to_num函数更改日期格式，代码如下：
df_arr=df.values
df_arr[:,0]=date_to_num(df_arr[:,0])
4. 使用python绘制股票k线图
使用mpl_finance库的candlestick_ochl函数，candlestick_ochl函数中的第一个参数ax表示画布子图的名称，第二个参数df_arr表示股价历史数据，第三个参数width表示k线柱的宽度，colorup表示收盘价高于开盘价时柱形的颜色，colordown表示收盘价低于开盘价时的柱形颜色（国内股价使用红色表示涨，绿色表示跌，美股相反因此需要说明这两个参数），alpha表示柱形的颜色透明度。
fig,ax=plt.subplots(figsize=(15,6))
mpf.candlestick_ochl(ax,df_arr,width=0.6,colorup='r',colordown='g',alpha=1)#绘制k线图
plt.grid(True)#绘制网格
ax.xaxis_date()#设置x轴的刻度格式为常规日期格式
绘制k线图如下：
5.添加均线图
为k线图分别添加5日均线和10日均线。
使用rolling函数和mean函数计算5日均线和10日均线数据，代码如下：
df['M5']=df['close'].rolling(5).mean()#5日均线
df['M10']=df['close'].rolling(10).mean()#10日均线
6.为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)#绘制k线图
plt.plot(df_arr[:,0],df['M5'])#绘制5日均线
plt.plot(df_arr[:,0],df['M10'])#绘制10日均线
plt.grid(True)#绘制网格线
plt.title('万科A')#设置标题
plt.xlabel('日期')#设置x轴标签
plt.ylabel('价格')#设置y轴标签
ax.xaxis_date()#设置x轴的刻度格式为常规日期格式
绘制如图所示：
