文章目录

一、绘制股票的数据

1.数据查看

import pandas as pd
pd.read_excel('/data/茅台股份.xlsx).head()

茅台股份数据从各大财经网上均可下载

2.绘制OHLC线图

OHLC是什么

  • O:代表开盘汇率Open
  • H:代表最高汇率High
  • L: 代表最低汇率Low
  • C: 代表收盘汇率Close

普通的OHLC图

import pandas as pd
import seaborn as sns
import matplotlib.pyplot as plt

#设置尺寸
plt.rcParams['figure.figsize'] = (18,6)

#设置中文问题
sns.set_style('darkgrid',{
   'font.sans-serif':['SimHei','Arial']})

#读数据
df_stock = pd.read_excel('./data/茅台股份.xlsx')

#抽取开盘，最高，最低，收盘 这些数据（就是讲dataframe变得小一点）
new_df = df_stock.loc[:, ['开盘','最高','最低','收盘']]

#绘制OHLC
plt.title('茅台股份的OHLC图')
sns.lineplot(data = new_df.iloc[:150])#前150行数据

#用这四列进行图形的绘制

通过K线图绘制OHLC图

上面的不太美观，我们通过k线图来画
找个k线的例子

这个图在matplotlib和sns中不自带，我们要怎么画呢？
要引入一个其他的包
pip install mpl_finance -i https://pypi.douban.com/simple
后面是豆瓣源下载的意思
绘制k线图的时候用吗candlesticj_ohlc()函数
参数：
candlesticj_ohlc(ax,quotes,width=.2,colorup='k',colordown='r',alpha=.1)
参数解释：

  • ax：图标的轴对象
  • quetes：股票数据，不能是一般的股票数据，必须是一个生成器的类型
  • colorup是设置涨的颜色
  • colordown是设置跌的颜色 
import pandas as pd
import matplotlib.pyplot as plt
import matplotlib.dates as dates
import mpl_finance as mf

#设置尺寸
plt.rcParams['figuer.figsize'] = (18,6)

#设置中文显示
plt.rcParamsp['font.family'] = ['SimHei']

#读数据
df_stock = pd.read_excel('./data/茅台股份.xlsx')

#把日期修改为行索引【重要】
df_stock.set_index('日期'，inplace = True)#inplace对数据进行永久改变

#抽取开盘，最高，最低，收盘 这些数据（就是讲dataframe变得小一点）
new_df = df_stock.loc[:, ['开盘','最高','最低','收盘']]
new_df = df_stock.iloc[:100]#取前100条来绘制

#重要的时间
#时间对象因为画图，必须要求是整数类型，用dates将时间转换为整数型
#转换后要压缩时间为可迭代类型
#to_pydatetime()转变为python当中datetime库下面的数据类型（时间元组）
zip_date = zip(dates.date2num(new.df, index.to_pydatetime()),new_df.开盘,new_df.最高，new_df.最低，new_df.收盘)

#获取画布, plt.gca直接生成画布
ax = plt.gca()

#绘制K线图
plt.title('茅台股份的OHLC图', size = 20)
mf.candlestick_ohlc(ax, zip_date, width = 1, colorup = 'r', colordown = 'g')

#把x轴的时间整数替换为时间日期
ax.axis_date()

#把x轴的文字倾斜
plt.xticks(rotation=45)