Python绘制专业的K线图 源代码解析

目录
  • 1、股票数据
  • 2、数据处理
  • 3、绘制K线
  • 4、去除图中非交易日
  • 5、在K线图中,添加成交量

K线图简介:

K线图又被成为“蜡烛图”、“阴阳线”等,它在视觉效果上可以很清晰得凸显出市场多空形势,K线图成为大家查看行情数据以及各式量化分析不可或缺的一环。在K线图常见的时间跨度分钟、日、周以及月。

K线由高开低收四个价格绘制而成。分为阳线与阴线两种,收盘价高于开盘价时为阳线,收盘价低于开盘价时为阴线;K线图的示意图如下:

K线由矩形实体与上下两根影线组成,实体上方的影线成为上影线,下方的成为下影线。实体与阴线相对长短,可形成多种形态。

1、股票数据

我们从恒有数金融数据社区,获取股票市场历史行情数据。我们获取2021年6月1号至2021年8月1号,恒生电子(600570.SH)的日行情数据,代码及执行结果如下。

# 加载取数与绘图所需的函数包
import pandas as pd
import datetime
from hs_udata import set_token,stock_quote_daily
from mpl_finance import candlestick_ohlc
import matplotlib as mpl
import matplotlib.pyplot as plt
import matplotlib.dates as mdates
mpl.rcParams['font.sans-serif'] = ['SimHei'] # 指定默认字体
mpl.rcParams['axes.unicode_minus'] = False  # 解决保存图像是负号'-'显示为方块的问题

def GetData(stock_code,start,end):
    #stock_code:获取股票数据的股票代码
    #     start:开始日期
    #       end:结束日期
    date_start=datetime.datetime.strptime(start,'%Y-%m-%d')
    date_end  =datetime.datetime.strptime(end,'%Y-%m-%d')
    data = pd.DataFrame([])
    while date_start<date_end:
        # 获取日行情数据,接口说明见 https://udata.hs.net/datas/332/
        # adjust_way枚举值为:0-不复权,1-前复权,2-后复权,此处取前复权
        data_i = stock_quote_daily(en_prod_code=stock_code
                                   ,trading_date=date_start.strftime('%Y%m%d')
                                   ,adjust_way = 1)
        data=pd.concat([data,data_i],axis=0)      # 将行情数据按行拼接
        date_start+=datetime.timedelta(days=1)    # 日期变量自增
    # 返回行情数据
    return data

#1、获取行情数据
stock_code = "600570.SH"     # 恒生电子 股票代码是600570.SH
start='2021-06-01'
end  ='2021-08-01'
set_token(token = 'xxxxxxxxxxxxxxxxxxxxxxxx')   # 注册后,获取并替换token
data = GetData(stock_code,start,end)
data

2、数据处理

由于恒有数的stock_quote_daily接口返回参数较多,包含了非交易日及停牌日期的数据。使用candlestick_ohlc包绘制K线图时,需要将日期转为数值。程序代码与执行结果如下:

#2、数据处理
data = data.loc[data.turnover_status=='交易']             # 剔除非交易日
data_price = data[['trading_date','open_price','high_price','low_price','close_price'
                   ,'business_amount']]                  # 选取日期、高开低收价格、成交量数据
data_price.set_index('trading_date', inplace=True)       # 将日期作为索引
data_price = data_price.astype(float)                    # 将价格数据类型转为浮点数
# 将日期格式转为 candlestick_ohlc 可识别的数值
data_price['Date'] = list(map(lambda x:mdates.date2num(datetime.datetime.strptime(x,'%Y-%m-%d')),data_price.index.tolist()))
data_price

3、绘制K线

使用mpl_finance函数包中candlestick_ohlc函数进行绘图,程序如下:

#3、绘制K线图
# 提取绘图数据
ohlc = data_price[['Date','open_price','high_price','low_price','close_price']]
f1, ax = plt.subplots(figsize = (12,6))                        # 创建图片
candlestick_ohlc(ax, ohlc.values.tolist(), width=.7
                 , colorup='red', colordown='green')           # 使用candlestick_ohlc绘图
ax.xaxis.set_major_formatter(mdates.DateFormatter('%Y-%m-%d')) # 设置横轴日期格式
plt.xticks(rotation=30)                                        # 日期显示的旋转角度
plt.title(stock_code,fontsize = 14)                            # 设置图片标题
plt.xlabel('日 期',fontsize = 14)                              # 设置横轴标题
plt.ylabel('价 格(元)',fontsize = 14)                        # 设置纵轴标题
plt.show()

4、去除图中非交易日

由于candlestick_ohlc函数默认绘制的K线图并未剔出非交易日(周末、节假日和停牌日期),导致K线之间存在空白间隔。下面我们通过修改绘图数据中横轴数据,修改横轴标注日期,实现剔除图中的非交易日数据。

# 4、去除非交易日的间隔
ohlc = data_price[['Date','open_price','high_price','low_price','close_price']]
ohlc.loc[:,'Date'] = range(len(ohlc))     # 重新赋值横轴数据,使横轴数据为连续数值
# 绘图
f1, ax = plt.subplots(figsize = (12,6))
candlestick_ohlc(ax, ohlc.values.tolist(), width=.7, colorup='red', colordown='green')
plt.xticks(rotation=30)                                        # 日期显示的旋转角度
plt.title(stock_code,fontsize = 14)                            # 设置图片标题
plt.xlabel('日 期',fontsize = 14)                               # 设置横轴标题
plt.ylabel('价 格(元)',fontsize = 14)                          # 设置纵轴标题
# 修改横轴标注日期
date_list = ohlc.index.tolist()           # 获取日期列表
xticks_len = round(len(date_list)/(len(ax.get_xticks())-1))    # 获取默认横轴标注的间隔
xticks_num = range(0,len(date_list),xticks_len)                # 生成横轴标注位置列表
xticks_str = list(map(lambda x:date_list[int(x)],xticks_num))  # 生成正在标注日期列表
ax.set_xticks(xticks_num)                                      # 设置横轴标注位置
ax.set_xticklabels(xticks_str)                                 # 设置横轴标注日期
plt.show()

5、在K线图中,添加成交量

K线图中,除了K线数据,一般还配有成交量数据。恒有数的stock_quote_daily接口返回的数据中,也有成交量数据。将K线图与成交量绘制在同一张图的程序如下:

#5、绘制成交量
fig = plt.figure(figsize=(12,10))
grid = plt.GridSpec(12, 10, wspace=0.5, hspace=0.5)
#(1)绘制K线图
# K线数据
ohlc = data_price[['Date','open_price','high_price','low_price','close_price']]
ohlc.loc[:,'Date'] = range(len(ohlc))     # 重新赋值横轴数据,绘制K线图无间隔
# 绘制K线
ax1 = fig.add_subplot(grid[0:8,0:12])   # 设置K线图的尺寸
candlestick_ohlc(ax1, ohlc.values.tolist(), width=.7
                 , colorup='red', colordown='green')
plt.title(stock_code,fontsize = 14)     # 设置图片标题
plt.ylabel('价 格(元)',fontsize = 14)   # 设置纵轴标题
ax1.set_xticks([])                      # 日期标注在成交量中,故清空此处x轴刻度
ax1.set_xticklabels([])                 # 日期标注在成交量中,故清空此处x轴
#(2)绘制成交量
# 成交量数据
data_volume = data_price[['Date','close_price','open_price','business_amount']]
data_volume['color'] = data_volume.apply(lambda row: 1 if row['close_price'] >= row['open_price'] else 0, axis=1)        # 计算成交量柱状图对应的颜色,使之与K线颜色一致
data_volume.Date = ohlc.Date
# 绘制成交量
ax2 = fig.add_subplot(grid[8:10,0:12])  # 设置成交量图形尺寸
ax2.bar(data_volume.query('color==1')['Date']
        , data_volume.query('color==1')['business_amount']
        , color='r')                    # 绘制红色柱状图
ax2.bar(data_volume.query('color==0')['Date']
        , data_volume.query('color==0')['business_amount']
        , color='g')                    # 绘制绿色柱状图
plt.xticks(rotation=30)
plt.xlabel('日 期',fontsize = 14)                               # 设置横轴标题
# 修改横轴日期标注
date_list = ohlc.index.tolist()           # 获取日期列表
xticks_len = round(len(date_list)/(len(ax2.get_xticks())-1))      # 获取默认横轴标注的间隔
xticks_num = range(0,len(date_list),xticks_len)                   # 生成横轴标注位置列表
xticks_str = list(map(lambda x:date_list[int(x)],xticks_num))     # 生成正在标注日期列表
ax2.set_xticks(xticks_num)                                        # 设置横轴标注位置
ax2.set_xticklabels(xticks_str)                                   # 设置横轴标注日期
plt.show()

到此这篇关于Python绘制专业的K线图 源代码解析的文章就介绍到这了,更多相关Python绘制专业的K线图 内容请搜索我们以前的文章或继续浏览下面的相关文章希望大家以后多多支持我们!

(0)

相关推荐

  • 使用Python画股票的K线图的方法步骤

    导言 本文简单介绍了如何从网易财经获取某支股票的价格数据,并根据价格数据画出相应的日K线图.有助于新手了解并使用Python的相关功能.包括列表.自定义函数.for循环.if函数以及如何使用matplotlib进行作图等内容. 第一步:从网易财经获取股票的价格数据 我一般是在网易财经查看某支股票的价格和成交数据,网易财经可以查到任意沪深的股票,我们使用招商银行的数据作为参考. 1.构建爬虫获取股票价格数据 这里不对Python做介绍了,如果需要了解什么是Python,可以自行百度或者访问Pyth

  • 利用python numpy+matplotlib绘制股票k线图的方法

    一.python numpy + matplotlib 画股票k线图 # -- coding: utf-8 -- import requests import numpy as np from matplotlib import pyplot as plt from matplotlib import animation fig = plt.figure(figsize=(8,6), dpi=72,facecolor="white") axes = plt.subplot(111) a

  • Python绘制K线图之可视化神器pyecharts的使用

    K线图 概念 股市及期货市bai场中的K线图的du画法包含四个zhi数据,即开盘dao价.最高价.最低价zhuan.收盘价,所有的shuk线都是围绕这四个数据展开,反映大势的状况和价格信息.如果把每日的K线图放在一张纸上,就能得到日K线图,同样也可画出周K线图.月K线图.研究金融的小伙伴肯定比较熟悉这个,那么我们看起来比较复杂的K线图,又是这样画出来的,本文我们将一起探索K线图的魅力与神奇之处吧! K线图 用处 K线图用处于股票分析,作为数据分析,以后的进入大数据肯定是一个趋势和热潮,K线图的专

  • Python绘制专业的K线图 源代码解析

    目录 1.股票数据 2.数据处理 3.绘制K线 4.去除图中非交易日 5.在K线图中,添加成交量 K线图简介: K线图又被成为"蜡烛图"."阴阳线"等,它在视觉效果上可以很清晰得凸显出市场多空形势,K线图成为大家查看行情数据以及各式量化分析不可或缺的一环.在K线图常见的时间跨度分钟.日.周以及月. K线由高开低收四个价格绘制而成.分为阳线与阴线两种,收盘价高于开盘价时为阳线,收盘价低于开盘价时为阴线:K线图的示意图如下: K线由矩形实体与上下两根影线组成,实体上方的

  • Python+Tkinter实现股票K线图的绘制

    目录 子窗口 子窗口框架 绘制K线图 在前面的文章中,我们一起学习了如何通过 Python 抓取东方财富网的实时股票数据以及如何制作成 Tkinter GUI 程序,链接如下 用 Python 爬取股票实时数据 Tkinter制作股票数据抓取小程序,有点秀! 今天我们就在这个基础上,在 Tkinter 程序中绘制 K 线图,一起来看看吧 子窗口 我们今天的整体代码还是基于上次的 Tkinter 股票程序,在主类 MainCreator 下面创建一个函数 create_subwindow     

  • 利用Matlab绘制优美的k线图

    目录 效果 代码 效果一 极光 效果二 暖调 效果三 黑白 本期又是一个花里胡哨的数据可视化,前两天刷到了耐克的视觉设计师Gladys Orteza绘制的k线图作品,把沉闷的股票图变成了精彩的风景,但是那些大部分是真的完全看不清,我这里挑选了几个能看清的k线图风格将其用MATLAB进行了实现. 效果 代码 代码中使用的数据 SimulatedStock.mat 是MATLAB自带的示例数据,因此不需要下载,不过要绘制k线图,仍需安装Financial Toolbox工具箱. 效果一 极光 % @

  • 使用PyQtGraph绘制精美的股票行情K线图的示例代码

    pyqtgraph是Python平台上一种功能强大的2D/3D绘图库,相对于matplotlib库,由于其在内部实现方式上,使用了高速计算的numpy信号处理库以及Qt的GraphicsView框架,因此它在大数据量的处理及快速显示方面有着天然的优势,非常适合于需要快速绘图更新.视频或实时交互性的操作场合,在数学.科学和工程领域都有着广泛的应用. K线图介绍 对于股票交易者来讲,K线图是弄清股票一段时间走势的一种最基本的图形工具,K线分为阳线和阴线,阳线和阴线都包含了开盘价.收盘价.最高价和最低

  • python爬虫爬取股票的k线图

    目录 前言 数据来源分析 数据抓取 总结 前言 之前已经讲述了一些关于 python 获取基金的一些信息,最近又有了一些新发现,和大家分享一下,这个是非常重要的内容,非常重要的内容.这个数据也是非常的敏感,在一些搞量化交易的平台上,这些数据都是要收费的,而且数据的质量也不能保障.这个内容就是如何获取股票交易的 k 线数据. 数据来源分析 我是非常欣赏东方某富的,因为同为券商,和别的公司确实不大一样,有这互联网的基因,可以这样说,是因为它的出现改变了一些行业的规则.话不多说,这里以海尔智家为例,抓

  • 基于Python轻松制作一个股票K线图网站

    目录 获取股票数据 PyEcharts 作图 构建 Web 框架 视图函数编写 模板编写 编辑主逻辑 前端页面编写 在前面的文章中,我们学习了如何使用 Tkinter 构建股票数据抓取以及展示K线图功能,虽然大致的功能已经具备,但是在当今这个人手一个 Web 服务的年代,GUI 程序还是没有 Web 服务来的香啊. 我们需要用到的知识包括 PyEcharts 的使用,tushare 库获取股票数据的方法以及 Flask 的基本用法. 获取股票数据 我们先来看下 tushare 的使用,这个应该是

  • 利用ECharts.js画K线图的方法示例

    前言 最近有一个统计的项目要做,在前端的数据需要用图表的形式展示.网上搜索了一下,发现有几种统计图库. MSChart 这个是Visual Studio里的自带控件,使用比较简单,不过数据这块需要在后台绑定. ichartjs 是一款基于HTML5的图形库.使用纯javascript语言, 利用HTML5的canvas标签绘制各式图形. 支持饼图.环形图.折线图.面积图.柱形图.条形图等. Chart.js 也是一款基于HTML5的图形库和ichartjs整体类似.不过Chart.js的教程文档

随机推荐