2597 lines
94 KiB
Python
2597 lines
94 KiB
Python
|
# -*- coding: utf-8 -*-
|
|||
|
|
|
|||
|
|
import os
|
|||
|
|
import sys
|
|||
|
|
import pickle
|
|||
|
|
import math
|
|||
|
|
import datetime
|
|||
|
|
import itertools
|
|||
|
|
import matplotlib
|
|||
|
|
|
|||
|
|
matplotlib.use("WXAgg", warn=True) # 这个要紧跟在 import matplotlib 之后，而且必须安装了 wxpython 2.8 才行。
|
|||
|
|
|
|||
|
|
import matplotlib.pyplot as pyplot
|
|||
|
|
import matplotlib.font_manager as font_manager
|
|||
|
|
|
|||
|
|
import numpy
|
|||
|
|
from matplotlib.ticker import NullLocator, FixedLocator, MultipleLocator, FuncFormatter, NullFormatter
|
|||
|
|
from matplotlib.patches import Ellipse
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
__font_properties__= font_manager.FontProperties(fname='/usr/share/fonts/truetype/wqy/wqy-zenhei.ttc')
|
|||
|
|
__color_lightsalmon__= '#ffa07a'
|
|||
|
|
__color_pink__= '#ffc0cb'
|
|||
|
|
__color_navy__= '#000080'
|
|||
|
|
__color_gold__= '#FDDB05'
|
|||
|
|
__color_gray30__= '0.3'
|
|||
|
|
__color_gray70__= '0.7'
|
|||
|
|
__color_lightblue__= 'lightblue'
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
__shrink__= 1.0 / 4
|
|||
|
|
__expbase__= 1.1
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
class SubPlot_BasicInfo:
|
|||
|
|
'''
|
|||
|
|
公司的基本信息
|
|||
|
|
Note: this is not "real" subplot, no Axes object contained.
|
|||
|
|
'''
|
|||
|
|
|
|||
|
|
def __init__(self, pdata, parent, name):
|
|||
|
|
self._name= name
|
|||
|
|
self._pdata= pdata
|
|||
|
|
self._cominfo= self._pdata[u'公司信息']
|
|||
|
|
self._parent= parent
|
|||
|
|
|
|||
|
|
self._Axes= None
|
|||
|
|
|
|||
|
|
self._xsize, \
|
|||
|
|
self._ysize= self._compute_size()
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
def _compute_size(self):
|
|||
|
|
return (300.0, 1.8)
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
def get_size(self):
|
|||
|
|
return (self._xsize, self._ysize)
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
def build_axes(self, figobj, rect):
|
|||
|
|
axes= figobj.add_axes(rect)
|
|||
|
|
axes.set_frame_on(False)
|
|||
|
|
self._Axes= axes
|
|||
|
|
|
|||
|
|
self.set_xticks()
|
|||
|
|
self.set_yticks()
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
def set_xticks(self):
|
|||
|
|
|
|||
|
|
axes= self._Axes
|
|||
|
|
xaxis= axes.get_xaxis()
|
|||
|
|
|
|||
|
|
# 设定 X 轴坐标的范围
|
|||
|
|
#==================================================================================================================================================
|
|||
|
|
axes.set_xlim(0, self._xsize)
|
|||
|
|
|
|||
|
|
xaxis.set_major_locator(NullLocator())
|
|||
|
|
|
|||
|
|
for mal in axes.get_xticklabels(minor=False):
|
|||
|
|
mal.set_visible(False)
|
|||
|
|
|
|||
|
|
for mil in axes.get_xticklabels(minor=True):
|
|||
|
|
mil.set_visible(False)
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
def set_yticks(self):
|
|||
|
|
|
|||
|
|
axes= self._Axes
|
|||
|
|
yaxis= axes.get_yaxis()
|
|||
|
|
|
|||
|
|
# 设定 X 轴坐标的范围
|
|||
|
|
#==================================================================================================================================================
|
|||
|
|
axes.set_ylim(0, self._ysize)
|
|||
|
|
|
|||
|
|
yaxis.set_major_locator(NullLocator())
|
|||
|
|
|
|||
|
|
for mal in axes.get_yticklabels(minor=False):
|
|||
|
|
mal.set_visible(False)
|
|||
|
|
|
|||
|
|
for mil in axes.get_yticklabels(minor=True):
|
|||
|
|
mil.set_visible(False)
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
def plot(self):
|
|||
|
|
|
|||
|
|
self.plot_codesymbol(xbase=0.0, ybase=self._ysize)
|
|||
|
|
self.plot_codesymbol_2(xbase=self._xsize, ybase=self._ysize)
|
|||
|
|
self.plot_companyname(xbase=0.0, ybase=self._ysize-0.8)
|
|||
|
|
self.plot_companylocation(xbase=48.0, ybase=self._ysize)
|
|||
|
|
self.plot_mainbusiness(xbase=48.0, ybase=self._ysize)
|
|||
|
|
self.plot_description(xbase=90.0, ybase=self._ysize)
|
|||
|
|
self.plot_sortinginfo(xbase=165.0, ybase=self._ysize)
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
def plot_codesymbol(self, xbase, ybase):
|
|||
|
|
'''
|
|||
|
|
交易代码、公司简称
|
|||
|
|
'''
|
|||
|
|
|
|||
|
|
txtstr= self._cominfo[u'代码'] + u' ' + self._cominfo[u'简称']
|
|||
|
|
label= self._Axes.text(xbase, ybase, txtstr, fontproperties=__font_properties__, verticalalignment='top', horizontalalignment='left')
|
|||
|
|
label.set_fontsize(16.0)
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
def plot_codesymbol_2(self, xbase, ybase):
|
|||
|
|
txtstr= self._cominfo[u'简称二']
|
|||
|
|
label= self._Axes.text(xbase, ybase, txtstr, fontproperties=__font_properties__, verticalalignment='top', horizontalalignment='right')
|
|||
|
|
label.set_fontsize(16.0)
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
def plot_companyname(self, xbase, ybase):
|
|||
|
|
'''
|
|||
|
|
曾用名、全名、英文名
|
|||
|
|
'''
|
|||
|
|
|
|||
|
|
txtstr= self._cominfo[u'基本情况'][u'曾用名']
|
|||
|
|
txtlist= txtstr.split('->')
|
|||
|
|
if len(txtlist) > 15:
|
|||
|
|
txtstr= ' -> '.join(txtlist[:5]) + ' ->\n' + ' -> '.join(txtlist[5:10]) + ' ->\n' + ' -> '.join(txtlist[10:15]) + ' ->\n' + ' -> '.join(txtlist[15:]) + '\n'
|
|||
|
|
elif len(txtlist) > 10:
|
|||
|
|
txtstr= ' -> '.join(txtlist[:5]) + ' ->\n' + ' -> '.join(txtlist[5:10]) + ' ->\n' + ' -> '.join(txtlist[10:]) + '\n'
|
|||
|
|
elif len(txtlist) > 5:
|
|||
|
|
txtstr= ' -> '.join(txtlist[:5]) + ' ->\n' + ' -> '.join(txtlist[5:]) + '\n'
|
|||
|
|
else:
|
|||
|
|
txtstr= ' -> '.join(txtlist) + '\n'
|
|||
|
|
txtstr += self._cominfo[u'基本情况'][u'公司名称'] + '\n'
|
|||
|
|
txtstr += self._cominfo[u'基本情况'][u'英文名称']
|
|||
|
|
|
|||
|
|
label= self._Axes.text(xbase, ybase, txtstr, fontproperties=__font_properties__, verticalalignment='top', horizontalalignment='left')
|
|||
|
|
label.set_fontsize(4.5)
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
def plot_companylocation(self, xbase, ybase):
|
|||
|
|
'''
|
|||
|
|
地域、所属行业、上市日期
|
|||
|
|
'''
|
|||
|
|
|
|||
|
|
txtstr= self._cominfo[u'公司概况'][u'区域'] + ' ' + self._cominfo[u'公司概况'][u'所属行业'] + ' ' + self._cominfo[u'发行相关'][u'上市日期']
|
|||
|
|
|
|||
|
|
label= self._Axes.text(xbase, ybase, txtstr, fontproperties=__font_properties__, verticalalignment='top', horizontalalignment='left')
|
|||
|
|
label.set_fontsize(6.5)
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
def plot_mainbusiness(self, xbase, ybase):
|
|||
|
|
'''
|
|||
|
|
主营业务
|
|||
|
|
'''
|
|||
|
|
# 查找表: (<文字长度>, <每行字数>, <字体大小>, <Y轴偏移量>)
|
|||
|
|
lookups= (
|
|||
|
|
(20, 10, 12.0, 0.5),
|
|||
|
|
(45, 15, 8.2, 0.5),
|
|||
|
|
(80, 20, 6.2, 0.5),
|
|||
|
|
(125, 25, 5.0, 0.5),
|
|||
|
|
(180, 30, 4.1, 0.5),
|
|||
|
|
(245, 35, 3.5, 0.4),
|
|||
|
|
(999999, 37, 3.4, 0.4)
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
txtstr= self._cominfo[u'基本情况'][u'主营业务']
|
|||
|
|
length= len(txtstr)
|
|||
|
|
for sizelimit, linelimit, fontsize, yshift in lookups:
|
|||
|
|
if length <= sizelimit:
|
|||
|
|
txtstr= '\n'.join([txtstr[linelimit*idx : linelimit*(idx+1)] for idx in range(length//linelimit + 1)])
|
|||
|
|
fsize= fontsize
|
|||
|
|
ycoord= ybase - yshift
|
|||
|
|
break
|
|||
|
|
|
|||
|
|
label= self._Axes.text(xbase, ycoord, txtstr, fontproperties=__font_properties__, verticalalignment='top', horizontalalignment='left', color='blue')
|
|||
|
|
label.set_fontsize(fsize)
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
def plot_description(self, xbase, ybase):
|
|||
|
|
'''
|
|||
|
|
公司简介
|
|||
|
|
'''
|
|||
|
|
# 查找表: (<文字长度>, <每行字数>, <字体大小>)
|
|||
|
|
lookups= (
|
|||
|
|
(150, 30, 7.0),
|
|||
|
|
(240, 40, 5.6),
|
|||
|
|
(329, 47, 4.8),
|
|||
|
|
(432, 54, 4.2),
|
|||
|
|
(576, 64, 3.5),
|
|||
|
|
(670, 67, 3.4),
|
|||
|
|
(792, 72, 3.1),
|
|||
|
|
(960, 80, 2.8),
|
|||
|
|
(1222, 94, 2.4),
|
|||
|
|
(1428, 102, 2.26),
|
|||
|
|
(1620, 108, 2.12),
|
|||
|
|
(1938, 114, 2.00),
|
|||
|
|
(999999, 130, 1.75)
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
txtstr= self._cominfo[u'公司概况'][u'公司简介'] # 26 ~ 2600 字符
|
|||
|
|
length= len(txtstr)
|
|||
|
|
|
|||
|
|
for sizelimit, linelimit, fontsize in lookups:
|
|||
|
|
if length <= sizelimit:
|
|||
|
|
txtstr= '\n'.join([txtstr[linelimit*idx : linelimit*(idx+1)] for idx in range(length//linelimit + 1)])
|
|||
|
|
fsize= fontsize
|
|||
|
|
break
|
|||
|
|
|
|||
|
|
label= self._Axes.text(xbase, ybase, txtstr, fontproperties=__font_properties__, verticalalignment='top', horizontalalignment='left')
|
|||
|
|
label.set_fontsize(fsize)
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
def plot_sortinginfo(self, xbase, ybase):
|
|||
|
|
'''
|
|||
|
|
行业板块信息
|
|||
|
|
'''
|
|||
|
|
infolist= self._cominfo[u'行业板块']
|
|||
|
|
|
|||
|
|
for idx in range(len(infolist)//10 + 1):
|
|||
|
|
txtstr= '\n'.join(infolist[10*idx : 10*(idx+1)])
|
|||
|
|
if not txtstr:
|
|||
|
|
break
|
|||
|
|
xcoord= xbase + 25.0*idx
|
|||
|
|
label= self._Axes.text(xcoord, ybase, txtstr, fontproperties=__font_properties__, verticalalignment='top', horizontalalignment='left', color='blue')
|
|||
|
|
label.set_fontsize(3.4)
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
class SubPlot_Financial:
|
|||
|
|
'''
|
|||
|
|
换手率子图
|
|||
|
|
'''
|
|||
|
|
pass
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
class SubPlot_PriceBase:
|
|||
|
|
'''
|
|||
|
|
|
|||
|
|
'''
|
|||
|
|
|
|||
|
|
def __init__(self, pdata, parent, xparams, name):
|
|||
|
|
'''
|
|||
|
|
|
|||
|
|
'''
|
|||
|
|
self._name= name # 派生类自己设置
|
|||
|
|
self._pdata= pdata
|
|||
|
|
self._parent= parent
|
|||
|
|
self._expbase= __expbase__
|
|||
|
|
self._xparams= xparams
|
|||
|
|
self._shrink= __shrink__ if name == 'pricefs' else 1.0
|
|||
|
|
|
|||
|
|
# 绘图数据
|
|||
|
|
quotes= pdata[u'行情']
|
|||
|
|
|
|||
|
|
if name == 'pricefs':
|
|||
|
|
self._dates= quotes[u'日期']
|
|||
|
|
self._open= quotes[u'开盘']
|
|||
|
|
self._close= quotes[u'收盘']
|
|||
|
|
self._high= quotes[u'最高']
|
|||
|
|
self._low= quotes[u'最低']
|
|||
|
|
if u'简化' in quotes: self._simple= quotes[u'简化']
|
|||
|
|
|
|||
|
|
# if u'换手率' in quotes: self._torate= quotes[u'换手率']
|
|||
|
|
# if u'成交量' in quotes: self._volume= quotes[u'成交量']
|
|||
|
|
# if u'成交额' in quotes: self._turnover= quotes[u'成交额']
|
|||
|
|
|
|||
|
|
if u'3日均' in quotes: self._average3= quotes[u'3日均']
|
|||
|
|
if u'5日均' in quotes: self._average5= quotes[u'5日均']
|
|||
|
|
if u'10日均' in quotes: self._average10= quotes[u'10日均']
|
|||
|
|
if u'30日均' in quotes: self._average30= quotes[u'30日均']
|
|||
|
|
if u'60日均' in quotes: self._average60= quotes[u'60日均']
|
|||
|
|
|
|||
|
|
if u'开盘二' in quotes:
|
|||
|
|
self._open_2= quotes[u'开盘二']
|
|||
|
|
self._close_2= quotes[u'收盘二']
|
|||
|
|
self._high_2= quotes[u'最高二']
|
|||
|
|
self._low_2= quotes[u'最低二']
|
|||
|
|
if u'简化二' in quotes: self._simple_2= quotes[u'简化二']
|
|||
|
|
|
|||
|
|
# if u'换手率二' in quotes: self._torate_2= quotes[u'换手率二']
|
|||
|
|
# if u'成交量二' in quotes: self._volume_2= quotes[u'成交量二']
|
|||
|
|
# if u'成交额二' in quotes: self._turnover_2= quotes[u'成交额二']
|
|||
|
|
|
|||
|
|
if u'3日均二' in quotes: self._average3_2= quotes[u'3日均二']
|
|||
|
|
if u'5日均二' in quotes: self._average5_2= quotes[u'5日均二']
|
|||
|
|
if u'10日均二' in quotes: self._average10_2= quotes[u'10日均二']
|
|||
|
|
if u'30日均二' in quotes: self._average30_2= quotes[u'30日均二']
|
|||
|
|
if u'60日均二' in quotes: self._average60_2= quotes[u'60日均二']
|
|||
|
|
|
|||
|
|
else:
|
|||
|
|
sidx, eidx= pdata[u'任务描述'][u'起始偏移'], pdata[u'任务描述'][u'结束偏移']
|
|||
|
|
|
|||
|
|
self._dates= quotes[u'日期'][sidx:eidx]
|
|||
|
|
self._open= quotes[u'开盘'][sidx:eidx]
|
|||
|
|
self._close= quotes[u'收盘'][sidx:eidx]
|
|||
|
|
self._high= quotes[u'最高'][sidx:eidx]
|
|||
|
|
self._low= quotes[u'最低'][sidx:eidx]
|
|||
|
|
if u'简化' in quotes: self._simple= quotes[u'简化'][sidx:eidx]
|
|||
|
|
|
|||
|
|
# if u'换手率' in quotes: self._torate= quotes[u'换手率'][sidx:eidx]
|
|||
|
|
# if u'成交量' in quotes: self._volume= quotes[u'成交量'][sidx:eidx]
|
|||
|
|
# if u'成交额' in quotes: self._turnover= quotes[u'成交额'][sidx:eidx]
|
|||
|
|
|
|||
|
|
if u'3日均' in quotes: self._average3= quotes[u'3日均'][sidx:eidx]
|
|||
|
|
if u'5日均' in quotes: self._average5= quotes[u'5日均'][sidx:eidx]
|
|||
|
|
if u'10日均' in quotes: self._average10= quotes[u'10日均'][sidx:eidx]
|
|||
|
|
if u'30日均' in quotes: self._average30= quotes[u'30日均'][sidx:eidx]
|
|||
|
|
if u'60日均' in quotes: self._average60= quotes[u'60日均'][sidx:eidx]
|
|||
|
|
|
|||
|
|
if u'开盘二' in quotes:
|
|||
|
|
self._open_2= quotes[u'开盘二'][sidx:eidx]
|
|||
|
|
self._close_2= quotes[u'收盘二'][sidx:eidx]
|
|||
|
|
self._high_2= quotes[u'最高二'][sidx:eidx]
|
|||
|
|
self._low_2= quotes[u'最低二'][sidx:eidx]
|
|||
|
|
if u'简化二' in quotes: self._simple_2= quotes[u'简化二'][sidx:eidx]
|
|||
|
|
|
|||
|
|
# if u'换手率二' in quotes: self._torate_2= quotes[u'换手率二'][sidx:eidx]
|
|||
|
|
# if u'成交量二' in quotes: self._volume_2= quotes[u'成交量二'][sidx:eidx]
|
|||
|
|
# if u'成交额二' in quotes: self._turnover_2= quotes[u'成交额二'][sidx:eidx]
|
|||
|
|
|
|||
|
|
if u'3日均二' in quotes: self._average3_2= quotes[u'3日均二'][sidx:eidx]
|
|||
|
|
if u'5日均二' in quotes: self._average5_2= quotes[u'5日均二'][sidx:eidx]
|
|||
|
|
if u'10日均二' in quotes: self._average10_2= quotes[u'10日均二'][sidx:eidx]
|
|||
|
|
if u'30日均二' in quotes: self._average30_2= quotes[u'30日均二'][sidx:eidx]
|
|||
|
|
if u'60日均二' in quotes: self._average60_2= quotes[u'60日均二'][sidx:eidx]
|
|||
|
|
|
|||
|
|
self._length= len(self._dates) # XXX: 由派生类设定
|
|||
|
|
|
|||
|
|
# 衍生数据
|
|||
|
|
#==============================================================================================================
|
|||
|
|
self._xindex= numpy.arange(self._length) # X 轴上的 index，一个辅助数据
|
|||
|
|
|
|||
|
|
self._zipoc= zip(self._open, self._close)
|
|||
|
|
self._up= numpy.array( [ True if po < pc and po is not None else False for po, pc in self._zipoc] ) # 标示出该天股价日内上涨的一个序列
|
|||
|
|
self._down= numpy.array( [ True if po > pc and po is not None else False for po, pc in self._zipoc] ) # 标示出该天股价日内下跌的一个序列
|
|||
|
|
self._side= numpy.array( [ True if po == pc and po is not None else False for po, pc in self._zipoc] ) # 标示出该天股价日内走平的一个序列
|
|||
|
|
|
|||
|
|
if u'开盘二' in quotes:
|
|||
|
|
self._zipoc_2= zip(self._open_2, self._close_2)
|
|||
|
|
self._up_2= numpy.array( [ True if po < pc and po is not None else False for po, pc in self._zipoc_2] ) # 标示出该天股价日内上涨的一个序列
|
|||
|
|
self._down_2= numpy.array( [ True if po > pc and po is not None else False for po, pc in self._zipoc_2] ) # 标示出该天股价日内下跌的一个序列
|
|||
|
|
self._side_2= numpy.array( [ True if po == pc and po is not None else False for po, pc in self._zipoc_2] ) # 标示出该天股价日内走平的一个序列
|
|||
|
|
|
|||
|
|
|
|||
|
|
self._Axes= None
|
|||
|
|
self._AxisX= None
|
|||
|
|
self._AxisY= None
|
|||
|
|
|
|||
|
|
self._xsize= 0.0
|
|||
|
|
self._ysize= 0.0
|
|||
|
|
|
|||
|
|
self._yhighlim= 0 # Y 轴最大坐标
|
|||
|
|
self._ylowlim= 0 # Y 轴最小坐标
|
|||
|
|
|
|||
|
|
if u'开盘二' in self._pdata[u'行情']:
|
|||
|
|
self._Axes_2= None # 如果有第二个行情数据，就建立另一个 Axes 对象
|
|||
|
|
self._AxisX_2= None
|
|||
|
|
self._AxisY_2= None
|
|||
|
|
|
|||
|
|
self._yhighlim_2= 0 # Y 轴最大坐标
|
|||
|
|
self._ylowlim_2= 0 # Y 轴最小坐标
|
|||
|
|
|
|||
|
|
self._compute_size()
|
|||
|
|
self._ytickset= self._compute_ytickset() # 需放在前一句后面
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
def _compute_size(self):
|
|||
|
|
'''
|
|||
|
|
根据绘图数据 pdata 计算出本子图的尺寸，修改数据成员
|
|||
|
|
'''
|
|||
|
|
quotes= self._pdata[u'行情']
|
|||
|
|
|
|||
|
|
popen= self._open[0] # int 类型
|
|||
|
|
|
|||
|
|
phigh= max( [ph for ph in self._high if ph is not None] ) # 最高价
|
|||
|
|
plow= min( [pl for pl in self._low if pl is not None] ) # 最低价
|
|||
|
|
|
|||
|
|
# Y 轴范围
|
|||
|
|
if self._name == 'pricefs':
|
|||
|
|
yhighlim= phigh * 1.2 # K线子图 Y 轴最大坐标
|
|||
|
|
ylowlim= plow / 1.2 # K线子图 Y 轴最小坐标
|
|||
|
|
else:
|
|||
|
|
yhighlim= phigh * 1.1 # K线子图 Y 轴最大坐标
|
|||
|
|
ylowlim= plow / 1.1 # K线子图 Y 轴最小坐标
|
|||
|
|
|
|||
|
|
self._yhighlim= yhighlim
|
|||
|
|
self._ylowlim= ylowlim
|
|||
|
|
|
|||
|
|
if u'开盘二' in quotes:
|
|||
|
|
popen_2= self._open_2[0] # 同上
|
|||
|
|
phigh_2= max( [ph for ph in self._high_2 if ph is not None] ) # 第二个行情的最高价
|
|||
|
|
phigh= max(phigh, int(phigh_2 * popen / float(popen_2))) # 以第一个行情为基准修正出的总最高价
|
|||
|
|
plow_2= min( [pl for pl in self._low_2 if pl is not None] ) # 最低价
|
|||
|
|
plow= min(plow, int(plow_2 * popen / float(popen_2))) # 以第一个行情为基准修正出的总最低价
|
|||
|
|
|
|||
|
|
if self._name == 'pricefs':
|
|||
|
|
yhighlim= phigh * 1.2 # K线子图 Y 轴最大坐标
|
|||
|
|
ylowlim= plow / 1.2 # K线子图 Y 轴最小坐标
|
|||
|
|
else:
|
|||
|
|
yhighlim= phigh * 1.1 # K线子图 Y 轴最大坐标
|
|||
|
|
ylowlim= plow / 1.1 # K线子图 Y 轴最小坐标
|
|||
|
|
|
|||
|
|
ylowlim_2= ylowlim * popen_2 / float(popen)
|
|||
|
|
yhighlim_2= yhighlim * popen_2 / float(popen)
|
|||
|
|
|
|||
|
|
self._yhighlim= yhighlim
|
|||
|
|
self._ylowlim= ylowlim
|
|||
|
|
|
|||
|
|
self._yhighlim_2= yhighlim_2
|
|||
|
|
self._ylowlim_2= ylowlim_2
|
|||
|
|
|
|||
|
|
# XXX: 价格在 Y 轴上的 “份数”。注意，虽然最高与最低价是以第一个行情为基准修正出来的，但其中包含的倍数因子对结果无影响，即:
|
|||
|
|
# log(base, num1) - log(base, num2) ==
|
|||
|
|
# log(base, num1/num2) ==
|
|||
|
|
# log(base, k*num1/k*num2) ==
|
|||
|
|
# log(base, k*num1) - log(base, k*num2)
|
|||
|
|
# ，这是对数运算的性质。
|
|||
|
|
xmargin= self._xparams['xmargin']
|
|||
|
|
self._xsize= (self._length + xmargin*2) * self._shrink # int, 所有数据的长度，就是天数
|
|||
|
|
self._ysize= (math.log(yhighlim, self._expbase) - math.log(ylowlim, self._expbase)) * self._shrink # float
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
def get_size(self):
|
|||
|
|
return (self._xsize, self._ysize)
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
def get_ylimits(self):
|
|||
|
|
return (self._yhighlim, self._ylowlim)
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
def build_axes(self, figobj, rect):
|
|||
|
|
'''
|
|||
|
|
初始化 self._Axes 对象
|
|||
|
|
'''
|
|||
|
|
# 添加 Axes 对象
|
|||
|
|
#==================================================================================================================================================
|
|||
|
|
if self._name == 'price' and 'torate' in self._parent._subplots:
|
|||
|
|
sharex= self._parent._subplots['torate'].get_axes()
|
|||
|
|
axes= figobj.add_axes(rect, axis_bgcolor='black', sharex=sharex)
|
|||
|
|
elif self._name == 'pricefs' and 'toratefs' in self._parent._subplots:
|
|||
|
|
sharex= self._parent._subplots['toratefs'].get_axes()
|
|||
|
|
axes= figobj.add_axes(rect, axis_bgcolor='black', sharex=sharex)
|
|||
|
|
else:
|
|||
|
|
axes= figobj.add_axes(rect, axis_bgcolor='black')
|
|||
|
|
|
|||
|
|
axes.set_axisbelow(True) # 网格线放在底层
|
|||
|
|
# axes.set_zorder(1) # XXX: 不顶用
|
|||
|
|
# axes.patch.set_visible(False) # hide the 'canvas'
|
|||
|
|
axes.set_yscale('log', basey=self._expbase) # 使用对数坐标
|
|||
|
|
|
|||
|
|
# 改变坐标线的颜色
|
|||
|
|
#==================================================================================================================================================
|
|||
|
|
for child in axes.get_children():
|
|||
|
|
if isinstance(child, matplotlib.spines.Spine):
|
|||
|
|
child.set_color(__color_gold__)
|
|||
|
|
|
|||
|
|
# 得到 X 轴 和 Y 轴 的两个 Axis 对象
|
|||
|
|
#==================================================================================================================================================
|
|||
|
|
xaxis= axes.get_xaxis()
|
|||
|
|
yaxis= axes.get_yaxis()
|
|||
|
|
|
|||
|
|
# 设置两个坐标轴上的网格线
|
|||
|
|
#==================================================================================================================================================
|
|||
|
|
xaxis.grid(True, 'major', color='0.3', linestyle='solid', linewidth=0.2)
|
|||
|
|
xaxis.grid(True, 'minor', color='0.3', linestyle='dotted', linewidth=0.1)
|
|||
|
|
|
|||
|
|
if self._name == 'pricefs': # 如果是小图，就不设辅助网格线
|
|||
|
|
yaxis.grid(True, 'major', color='0.3', linestyle='solid', linewidth=0.1)
|
|||
|
|
else:
|
|||
|
|
yaxis.grid(True, 'major', color='0.3', linestyle='solid', linewidth=0.2)
|
|||
|
|
yaxis.grid(True, 'minor', color='0.3', linestyle='dotted', linewidth=0.1)
|
|||
|
|
|
|||
|
|
yaxis.set_label_position('left')
|
|||
|
|
|
|||
|
|
self._Axes= axes
|
|||
|
|
self._AxisX= xaxis
|
|||
|
|
self._AxisY= yaxis
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
if u'开盘二' in self._pdata[u'行情']:
|
|||
|
|
# 添加 Axes 对象。注意，设置 axes_2 而不是 axes 的网格线，从而不会跑到 axes 边框上边的做法不顶用。
|
|||
|
|
#==================================================================================================================================================
|
|||
|
|
axes_2= axes.twinx() # twinx is problematic, no use no more.
|
|||
|
|
|
|||
|
|
# XXX: 下面这三行把第一个 axes 放在上面，这样不会被第二个 axes 的图形遮盖。用 zorder 不顶用。
|
|||
|
|
axes.figure.axes[-2:]= [axes_2, axes] # XXX:
|
|||
|
|
axes.set_frame_on(False) # 如果不做此设定，axes_2 的内容会看不见
|
|||
|
|
axes_2.set_frame_on(True)
|
|||
|
|
|
|||
|
|
axes_2.set_axis_bgcolor('black')
|
|||
|
|
axes_2.set_axisbelow(True) # 网格线放在底层
|
|||
|
|
axes_2.set_yscale('log', basey=self._expbase) # 使用对数坐标
|
|||
|
|
|
|||
|
|
# 改变坐标线的颜色
|
|||
|
|
#==================================================================================================================================================
|
|||
|
|
for child in axes_2.get_children():
|
|||
|
|
if isinstance(child, matplotlib.spines.Spine):
|
|||
|
|
child.set_color(__color_gold__)
|
|||
|
|
|
|||
|
|
# 得到 X 轴 和 Y 轴 的两个 Axis 对象
|
|||
|
|
#==================================================================================================================================================
|
|||
|
|
xaxis_2= axes_2.get_xaxis()
|
|||
|
|
yaxis_2= axes_2.get_yaxis()
|
|||
|
|
|
|||
|
|
# 设置两个坐标轴上的网格线
|
|||
|
|
#==================================================================================================================================================
|
|||
|
|
# xaxis_2.grid(True, 'major', color='0.3', linestyle='solid', linewidth=0.2)
|
|||
|
|
# xaxis_2.grid(True, 'minor', color='0.3', linestyle='dotted', linewidth=0.1)
|
|||
|
|
|
|||
|
|
# if self._name == 'pricefs': # 如果是小图，就不设辅助网格线
|
|||
|
|
# yaxis_2.grid(True, 'major', color='0.3', linestyle='solid', linewidth=0.1)
|
|||
|
|
# else:
|
|||
|
|
# yaxis_2.grid(True, 'major', color='0.3', linestyle='solid', linewidth=0.2)
|
|||
|
|
# yaxis_2.grid(True, 'minor', color='0.3', linestyle='dotted', linewidth=0.1)
|
|||
|
|
|
|||
|
|
yaxis_2.set_label_position('right')
|
|||
|
|
|
|||
|
|
self._Axes_2= axes_2
|
|||
|
|
self._AxisX_2= xaxis_2
|
|||
|
|
self._AxisY_2= yaxis_2
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
def set_xticks(self):
|
|||
|
|
|
|||
|
|
xMajorLocator= self._xparams['xMajorLocator']
|
|||
|
|
xMinorLocator= self._xparams['xMinorLocator']
|
|||
|
|
|
|||
|
|
axes= self._Axes
|
|||
|
|
xaxis= self._AxisX
|
|||
|
|
|
|||
|
|
# 设定 X 轴坐标的范围
|
|||
|
|
#==================================================================================================================================================
|
|||
|
|
xmargin= self._xparams['xmargin']
|
|||
|
|
axes.set_xlim(-xmargin, self._length + xmargin)
|
|||
|
|
|
|||
|
|
# 先设置 label 位置，再将 X 轴上的坐标设为不可见。因为与 成交量子图 共用 X 轴
|
|||
|
|
#==================================================================================================================================================
|
|||
|
|
|
|||
|
|
# 设定 X 轴的 Locator 和 Formatter
|
|||
|
|
xaxis.set_major_locator(xMajorLocator)
|
|||
|
|
# xaxis.set_major_formatter(xMajorFormatter)
|
|||
|
|
|
|||
|
|
xaxis.set_minor_locator(xMinorLocator)
|
|||
|
|
# xaxis.set_minor_formatter(xMinorFormatter)
|
|||
|
|
|
|||
|
|
# 将 X 轴上的坐标设为不可见。
|
|||
|
|
for mal in axes.get_xticklabels(minor=False):
|
|||
|
|
mal.set_visible(False)
|
|||
|
|
|
|||
|
|
for mil in axes.get_xticklabels(minor=True):
|
|||
|
|
mil.set_visible(False)
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
def set_xticks_2(self):
|
|||
|
|
quotes= self._pdata[u'行情']
|
|||
|
|
|
|||
|
|
axes= self._Axes_2
|
|||
|
|
xaxis= self._AxisX_2
|
|||
|
|
|
|||
|
|
xMajorLocator= self._xparams['xMajorLocator']
|
|||
|
|
xMinorLocator= self._xparams['xMinorLocator']
|
|||
|
|
|
|||
|
|
# 设定 X 轴坐标的范围
|
|||
|
|
#==================================================================================================================================================
|
|||
|
|
xmargin= self._xparams['xmargin']
|
|||
|
|
axes.set_xlim(-xmargin, self._length + xmargin)
|
|||
|
|
|
|||
|
|
# 先设置 label 位置，再将 X 轴上的坐标设为不可见。因为与 成交量子图 共用 X 轴
|
|||
|
|
#==================================================================================================================================================
|
|||
|
|
|
|||
|
|
# 设定 X 轴的 Locator 和 Formatter
|
|||
|
|
xaxis.set_major_locator(xMajorLocator)
|
|||
|
|
# xaxis.set_major_formatter(xMajorFormatter)
|
|||
|
|
|
|||
|
|
xaxis.set_minor_locator(xMinorLocator)
|
|||
|
|
# xaxis.set_minor_formatter(xMinorFormatter)
|
|||
|
|
|
|||
|
|
# 将 X 轴上的坐标设为不可见。
|
|||
|
|
for mal in axes.get_xticklabels(minor=False):
|
|||
|
|
mal.set_visible(False)
|
|||
|
|
|
|||
|
|
for mil in axes.get_xticklabels(minor=True):
|
|||
|
|
mil.set_visible(False)
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
def _compute_ytickset(self):
|
|||
|
|
'''
|
|||
|
|
计算 Y 轴坐标点的位置，包括第二个行情
|
|||
|
|
'''
|
|||
|
|
quotes= self._pdata[u'行情']
|
|||
|
|
expbase= self._expbase
|
|||
|
|
|
|||
|
|
ytickset= {}
|
|||
|
|
|
|||
|
|
yhighlim= self._yhighlim
|
|||
|
|
ylowlim= self._ylowlim
|
|||
|
|
|
|||
|
|
if u'开盘二' in quotes:
|
|||
|
|
yhighlim_2= self._yhighlim_2
|
|||
|
|
ylowlim_2= self._ylowlim_2
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
if self._name == 'price' and 'pricefs' in self._parent._subplots:
|
|||
|
|
tsetfs= self._parent._subplots['pricefs'].get_ytickset()
|
|||
|
|
|
|||
|
|
majors= tsetfs['major']
|
|||
|
|
while majors[-1] < yhighlim: majors.append(majors[-1] * expbase)
|
|||
|
|
while majors[0] > ylowlim: majors.insert(0, majors[0] / expbase)
|
|||
|
|
|
|||
|
|
minors= tsetfs['minor']
|
|||
|
|
while minors[-1] < yhighlim: minors.append(minors[-1] * expbase)
|
|||
|
|
while minors[0] > ylowlim: minors.insert(0, minors[0] / expbase)
|
|||
|
|
|
|||
|
|
ytickset['major']= [loc for loc in majors if loc > ylowlim and loc < yhighlim]
|
|||
|
|
ytickset['minor']= [loc for loc in minors if loc > ylowlim and loc < yhighlim]
|
|||
|
|
|
|||
|
|
else:
|
|||
|
|
|
|||
|
|
# 主要坐标点
|
|||
|
|
#----------------------------------------------------------------------------
|
|||
|
|
majors= [ylowlim]
|
|||
|
|
while majors[-1] < yhighlim: majors.append(majors[-1] * 1.1)
|
|||
|
|
|
|||
|
|
# 辅助坐标点
|
|||
|
|
#----------------------------------------------------------------------------
|
|||
|
|
minors= [ylowlim * 1.1**0.5]
|
|||
|
|
while minors[-1] < yhighlim: minors.append(minors[-1] * 1.1)
|
|||
|
|
|
|||
|
|
ytickset['major']= [loc for loc in majors if loc > ylowlim and loc < yhighlim] # 注意，第一项（ylowlim）被排除掉了
|
|||
|
|
ytickset['minor']= [loc for loc in minors if loc > ylowlim and loc < yhighlim]
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
if u'开盘二' in quotes:
|
|||
|
|
popen= self._open[0] # int 类型
|
|||
|
|
popen_2= self._open_2[0] # 同上
|
|||
|
|
|
|||
|
|
ytickset['major_2']= [loc * popen_2 / popen for loc in ytickset['major']]
|
|||
|
|
ytickset['minor_2']= [loc * popen_2 / popen for loc in ytickset['minor']]
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
return ytickset
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
def get_ytickset(self):
|
|||
|
|
return self._ytickset
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
def set_yticks(self):
|
|||
|
|
'''
|
|||
|
|
设置第一只行情的 Y 轴坐标，包括坐标值在图中间的显示
|
|||
|
|
'''
|
|||
|
|
|
|||
|
|
axes= self._Axes
|
|||
|
|
ylowlim= self._ylowlim
|
|||
|
|
yhighlim= self._yhighlim
|
|||
|
|
yaxis= self._AxisY
|
|||
|
|
|
|||
|
|
majorticks= self._ytickset['major']
|
|||
|
|
minorticks= self._ytickset['minor']
|
|||
|
|
|
|||
|
|
# 设定 Y 轴坐标的范围
|
|||
|
|
#==================================================================================================================================================
|
|||
|
|
axes.set_ylim(ylowlim, yhighlim)
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
# 设定 Y 轴上的坐标
|
|||
|
|
#==================================================================================================================================================
|
|||
|
|
|
|||
|
|
# 主要坐标点
|
|||
|
|
#----------------------------------------------------------------------------
|
|||
|
|
|
|||
|
|
yMajorLocator= FixedLocator(numpy.array(majorticks))
|
|||
|
|
|
|||
|
|
# 确定 Y 轴的 MajorFormatter
|
|||
|
|
def y_major_formatter(num, pos=None):
|
|||
|
|
return str(round(num/1000.0, 2))
|
|||
|
|
|
|||
|
|
yMajorFormatter= FuncFormatter(y_major_formatter)
|
|||
|
|
|
|||
|
|
# 设定 X 轴的 Locator 和 Formatter
|
|||
|
|
yaxis.set_major_locator(yMajorLocator)
|
|||
|
|
yaxis.set_major_formatter(yMajorFormatter)
|
|||
|
|
|
|||
|
|
# 设定 Y 轴主要坐标点与辅助坐标点的样式
|
|||
|
|
fsize= 4 if self._name == 'pricefs' else 6
|
|||
|
|
|
|||
|
|
for mal in axes.get_yticklabels(minor=False):
|
|||
|
|
mal.set_fontsize(fsize)
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
# 辅助坐标点
|
|||
|
|
#----------------------------------------------------------------------------
|
|||
|
|
|
|||
|
|
yMinorLocator= FixedLocator(numpy.array(minorticks))
|
|||
|
|
|
|||
|
|
# 确定 Y 轴的 MinorFormatter
|
|||
|
|
def y_minor_formatter(num, pos=None):
|
|||
|
|
return str(round(num/1000.0, 2))
|
|||
|
|
|
|||
|
|
yMinorFormatter= FuncFormatter(y_minor_formatter)
|
|||
|
|
|
|||
|
|
# 设定 X 轴的 Locator 和 Formatter
|
|||
|
|
yaxis.set_minor_locator(yMinorLocator)
|
|||
|
|
yaxis.set_minor_formatter(yMinorFormatter)
|
|||
|
|
|
|||
|
|
# 设定 Y 轴辅助坐标点的样式
|
|||
|
|
if self._name == 'pricefs':
|
|||
|
|
for mil in axes.get_yticklabels(minor=True):
|
|||
|
|
mil.set_visible(False)
|
|||
|
|
else:
|
|||
|
|
for mil in axes.get_yticklabels(minor=True):
|
|||
|
|
mil.set_fontsize(5)
|
|||
|
|
mil.set_color('blue')
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
def set_yticks_2(self):
|
|||
|
|
'''
|
|||
|
|
子图右侧的 Y 轴坐标
|
|||
|
|
'''
|
|||
|
|
|
|||
|
|
axes= self._Axes_2
|
|||
|
|
yaxis= self._AxisY_2
|
|||
|
|
|
|||
|
|
yhighlim_2= self._yhighlim_2
|
|||
|
|
ylowlim_2= self._ylowlim_2
|
|||
|
|
|
|||
|
|
majorticks_2= self._ytickset['major_2']
|
|||
|
|
minorticks_2= self._ytickset['minor_2']
|
|||
|
|
|
|||
|
|
# 设定 Y 轴坐标的范围
|
|||
|
|
#==================================================================================================================================================
|
|||
|
|
|
|||
|
|
axes.set_ylim(ylowlim_2, yhighlim_2)
|
|||
|
|
|
|||
|
|
# 设定 Y 轴上的坐标
|
|||
|
|
#==================================================================================================================================================
|
|||
|
|
|
|||
|
|
# 主要坐标点
|
|||
|
|
#----------------------------------------------------------------------------
|
|||
|
|
|
|||
|
|
yMajorLocator= FixedLocator(numpy.array(majorticks_2))
|
|||
|
|
|
|||
|
|
# 确定 Y 轴的 MajorFormatter
|
|||
|
|
def y_major_formatter(num, pos=None):
|
|||
|
|
return str(round(num/1000.0, 2))
|
|||
|
|
|
|||
|
|
yMajorFormatter= FuncFormatter(y_major_formatter)
|
|||
|
|
|
|||
|
|
# 设定 X 轴的 Locator 和 Formatter
|
|||
|
|
yaxis.set_major_locator(yMajorLocator)
|
|||
|
|
yaxis.set_major_formatter(yMajorFormatter)
|
|||
|
|
|
|||
|
|
# 设定 Y 轴主要坐标点与辅助坐标点的样式
|
|||
|
|
fsize= 4 if self._name == 'pricefs' else 6
|
|||
|
|
|
|||
|
|
for mal in axes.get_yticklabels(minor=False):
|
|||
|
|
mal.set_fontsize(fsize)
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
# 辅助坐标点
|
|||
|
|
#----------------------------------------------------------------------------
|
|||
|
|
|
|||
|
|
yMinorLocator= FixedLocator(numpy.array(minorticks_2)) # XXX minor ticks 已经在上面一并设置，这里不需要了。
|
|||
|
|
|
|||
|
|
# 确定 Y 轴的 MinorFormatter
|
|||
|
|
def y_minor_formatter(num, pos=None):
|
|||
|
|
return str(round(num/1000.0, 2))
|
|||
|
|
|
|||
|
|
yMinorFormatter= FuncFormatter(y_minor_formatter)
|
|||
|
|
|
|||
|
|
# 设定 X 轴的 Locator 和 Formatter
|
|||
|
|
yaxis.set_minor_locator(yMinorLocator)
|
|||
|
|
yaxis.set_minor_formatter(yMinorFormatter)
|
|||
|
|
# 设定 Y 轴主要坐标点与辅助坐标点的样式
|
|||
|
|
if self._name == 'pricefs':
|
|||
|
|
for mil in axes.get_yticklabels(minor=True):
|
|||
|
|
mil.set_visible(False)
|
|||
|
|
|
|||
|
|
else:
|
|||
|
|
for mil in axes.get_yticklabels(minor=True):
|
|||
|
|
mil.set_fontsize(5)
|
|||
|
|
mil.set_color('blue')
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
def plot(self):
|
|||
|
|
'''
|
|||
|
|
由派生类自己定义
|
|||
|
|
'''
|
|||
|
|
pass
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
def plot_candlestick(self):
|
|||
|
|
'''
|
|||
|
|
绘制 K 线
|
|||
|
|
'''
|
|||
|
|
axes= self._Axes
|
|||
|
|
|
|||
|
|
xindex= self._xindex
|
|||
|
|
|
|||
|
|
up= self._up
|
|||
|
|
down= self._down
|
|||
|
|
side= self._side
|
|||
|
|
|
|||
|
|
# 绘制 K 线部分
|
|||
|
|
#==================================================================================================================================================
|
|||
|
|
|
|||
|
|
# 对开收盘价进行视觉修正
|
|||
|
|
for idx, poc in enumerate(self._zipoc):
|
|||
|
|
if poc[0] == poc[1] and None not in poc:
|
|||
|
|
variant= int(round((poc[1]+1000)/2000.0, 0))
|
|||
|
|
self._open[idx]= poc[0] - variant # 稍微偏离一点，使得在图线上不致于完全看不到
|
|||
|
|
self._close[idx]= poc[1] + variant
|
|||
|
|
|
|||
|
|
rarray_open= numpy.array(self._open)
|
|||
|
|
rarray_close= numpy.array(self._close)
|
|||
|
|
rarray_high= numpy.array(self._high)
|
|||
|
|
rarray_low= numpy.array(self._low)
|
|||
|
|
|
|||
|
|
# XXX: 如果 up, down, side 里有一个全部为 False 组成，那么 vlines() 会报错。
|
|||
|
|
# XXX: 可以使用 alpha 参数调节透明度
|
|||
|
|
if True in up:
|
|||
|
|
axes.vlines(xindex[up], rarray_low[up], rarray_high[up], edgecolor='red', linewidth=0.6, label='_nolegend_', alpha=0.5)
|
|||
|
|
axes.vlines(xindex[up], rarray_open[up], rarray_close[up], edgecolor='red', linewidth=3.0, label='_nolegend_', alpha=0.5)
|
|||
|
|
|
|||
|
|
if True in down:
|
|||
|
|
axes.vlines(xindex[down], rarray_low[down], rarray_high[down], edgecolor='green', linewidth=0.6, label='_nolegend_', alpha=0.5)
|
|||
|
|
axes.vlines(xindex[down], rarray_open[down], rarray_close[down], edgecolor='green', linewidth=3.0, label='_nolegend_', alpha=0.5)
|
|||
|
|
|
|||
|
|
if True in side:
|
|||
|
|
axes.vlines(xindex[side], rarray_low[side], rarray_high[side], edgecolor='0.7', linewidth=0.6, label='_nolegend_', alpha=0.5)
|
|||
|
|
axes.vlines(xindex[side], rarray_open[side], rarray_close[side], edgecolor='0.7', linewidth=3.0, label='_nolegend_', alpha=0.5)
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
def plot_simplified(self):
|
|||
|
|
'''
|
|||
|
|
绘制简化行情
|
|||
|
|
'''
|
|||
|
|
xindex= self._xindex
|
|||
|
|
axes= self._Axes
|
|||
|
|
|
|||
|
|
rarray_simple= numpy.array(self._simple)
|
|||
|
|
axes.plot(xindex, rarray_simple, 'o-', color='white', linewidth=0.3, label='simple', \
|
|||
|
|
markersize=0.3, markeredgecolor='white', markeredgewidth=0.1, alpha=0.3) # 简化行情
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
def plot_average(self):
|
|||
|
|
'''
|
|||
|
|
绘制均线
|
|||
|
|
'''
|
|||
|
|
# 绘制均线部分
|
|||
|
|
#==================================================================================================================================================
|
|||
|
|
quotes= self._pdata[u'行情']
|
|||
|
|
|
|||
|
|
xindex= self._xindex
|
|||
|
|
axes= self._Axes
|
|||
|
|
|
|||
|
|
if self._name == 'pricefs':
|
|||
|
|
widthw= 0.1
|
|||
|
|
widthn= 0.07
|
|||
|
|
mksize= 0.07
|
|||
|
|
mkwidth= 0.1
|
|||
|
|
alpha= 1.0
|
|||
|
|
else:
|
|||
|
|
widthw= 0.2
|
|||
|
|
widthn= 0.1
|
|||
|
|
mksize= 0.3
|
|||
|
|
mkwidth= 0.1
|
|||
|
|
alpha= 1.0
|
|||
|
|
|
|||
|
|
if u'3日均' in quotes:
|
|||
|
|
rarray_3dayave= numpy.array(self._average3)
|
|||
|
|
axes.plot(xindex, rarray_3dayave, 'o-', color='white', linewidth=widthw, label='avg_3', \
|
|||
|
|
markersize=mksize, markeredgecolor='white', markeredgewidth=mkwidth, alpha=alpha) # 3日均线
|
|||
|
|
|
|||
|
|
if u'5日均' in quotes:
|
|||
|
|
rarray_5dayave= numpy.array(self._average5)
|
|||
|
|
axes.plot(xindex, rarray_5dayave, 'o-', color='white', linewidth=widthn, label='avg_5', \
|
|||
|
|
markersize=mksize, markeredgecolor='white', markeredgewidth=mkwidth, alpha=alpha) # 5日均线
|
|||
|
|
|
|||
|
|
if u'10日均' in quotes:
|
|||
|
|
rarray_10dayave= numpy.array(self._average10)
|
|||
|
|
axes.plot(xindex, rarray_10dayave, 'o-', color='yellow', linewidth=widthn, label='avg_10', \
|
|||
|
|
markersize=mksize, markeredgecolor='yellow', markeredgewidth=mkwidth, alpha=alpha) # 10日均线
|
|||
|
|
|
|||
|
|
if u'30日均' in quotes:
|
|||
|
|
rarray_30dayave= numpy.array(self._average30)
|
|||
|
|
axes.plot(xindex, rarray_30dayave, 'o-', color='cyan', linewidth=widthn, label='avg_30', \
|
|||
|
|
markersize=mksize, markeredgecolor='cyan', markeredgewidth=mkwidth, alpha=alpha) # 30日均线
|
|||
|
|
|
|||
|
|
if u'60日均' in quotes:
|
|||
|
|
rarray_60dayave= numpy.array(self._average60)
|
|||
|
|
axes.plot(xindex, rarray_60dayave, 'o-', color='magenta', linewidth=widthn, label='avg_60', \
|
|||
|
|
markersize=mksize, markeredgecolor='magenta', markeredgewidth=mkwidth, alpha=alpha) # 60日均线
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
def plot_adjustnotes(self):
|
|||
|
|
'''
|
|||
|
|
绘制复权提示
|
|||
|
|
'''
|
|||
|
|
quotes= self._pdata[u'行情']
|
|||
|
|
|
|||
|
|
firstday= self._dates[0]
|
|||
|
|
lastday= self._dates[-1]
|
|||
|
|
ylowlim= self._ylowlim
|
|||
|
|
yhighlim= self._yhighlim
|
|||
|
|
axes= self._Axes
|
|||
|
|
|
|||
|
|
# 使用 annotate() 进行标注。不用了，留作纪念。
|
|||
|
|
#===========================================================================================================================
|
|||
|
|
# adjdict= dict(quotes[u'相对复权']) # key 是 date string，value 是相对复权因子（float 类型）
|
|||
|
|
# el= Ellipse((2, -1), 0.5, 0.5)
|
|||
|
|
# for idx, dstr in enumerate(self._dates):
|
|||
|
|
# if dstr in adjdict:
|
|||
|
|
# axes.plot([idx, idx], [ylowlim, yhighlim], '-', color='purple', linewidth=0.1)
|
|||
|
|
# axes.annotate(u'复权\n' + str(adjdict[dstr]),
|
|||
|
|
# fontproperties=__font_properties__,
|
|||
|
|
# xy=(idx, yhighlim/1.1), xycoords='data',
|
|||
|
|
# xytext=(10, 5), textcoords='offset points', size=5, verticalalignment="center",
|
|||
|
|
# bbox=dict(boxstyle="round", facecolor='white', edgecolor='purple'),
|
|||
|
|
# arrowprops=dict(arrowstyle="wedge,tail_width=1.",
|
|||
|
|
# facecolor='white', edgecolor='purple',
|
|||
|
|
# patchA=None,
|
|||
|
|
# patchB=el,
|
|||
|
|
# relpos=(0.2, 0.8),
|
|||
|
|
# connectionstyle="arc3,rad=-0.1"),
|
|||
|
|
# alpha=0.5
|
|||
|
|
# )
|
|||
|
|
|
|||
|
|
adjrecs= [rec for rec in quotes[u'相对复权'] if rec[0] >= firstday and rec[0] <= lastday]
|
|||
|
|
|
|||
|
|
if self._name == 'pricefs':
|
|||
|
|
fsize= 3.0
|
|||
|
|
ycoord= yhighlim/1.3
|
|||
|
|
alpha= 1.0
|
|||
|
|
else:
|
|||
|
|
fsize= 5.0
|
|||
|
|
ycoord= yhighlim/1.12
|
|||
|
|
alpha= 1.0
|
|||
|
|
|
|||
|
|
for dstr, afac in adjrecs:
|
|||
|
|
idx= self._dates.index(dstr)
|
|||
|
|
axes.plot([idx, idx], [ylowlim, yhighlim], '-', color='purple', linewidth=0.1)
|
|||
|
|
label= axes.text( \
|
|||
|
|
idx, ycoord, \
|
|||
|
|
u'复权 ' + str(afac) + u'\n' + dstr, \
|
|||
|
|
fontproperties=__font_properties__, \
|
|||
|
|
horizontalalignment='left', \
|
|||
|
|
verticalalignment='top', \
|
|||
|
|
color='purple', \
|
|||
|
|
alpha=alpha
|
|||
|
|
)
|
|||
|
|
label.set_fontsize(fsize)
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
def plot_capchangenotes(self):
|
|||
|
|
'''
|
|||
|
|
绘制流通股本变更提示
|
|||
|
|
注意两个问题:
|
|||
|
|
1. 流通股本变更提示中的日期可能不是交易日期
|
|||
|
|
2. 流通股本变更提示涵盖个股的所有历史，有些内容可能在绘图目标区间以外
|
|||
|
|
'''
|
|||
|
|
pdata= self._pdata
|
|||
|
|
axes= self._Axes
|
|||
|
|
ylowlim= self._ylowlim
|
|||
|
|
yhighlim= self._yhighlim
|
|||
|
|
|
|||
|
|
firstday= self._dates[0]
|
|||
|
|
lastday= self._dates[-1]
|
|||
|
|
|
|||
|
|
# 把目标区间之外的记录滤掉
|
|||
|
|
changerecs= [rec for rec in pdata[u'公司信息'][u'流通股变更'] if rec[u'变更日期'] >= firstday and rec[u'变更日期'] <= lastday]
|
|||
|
|
|
|||
|
|
# 使用 annotate() 进行标注。不用了，留作纪念。
|
|||
|
|
#===========================================================================================================================
|
|||
|
|
# el= Ellipse((2, -1), 0.5, 0.5)
|
|||
|
|
# for datestr, chrate in changerecs:
|
|||
|
|
# dstr= [ds for ds in self._dates if ds >= datestr][0]
|
|||
|
|
# idx= self._dates.index(dstr)
|
|||
|
|
# axes.plot([idx, idx], [ylowlim, yhighlim], '-', color='yellow', linewidth=0.1)
|
|||
|
|
# axes.annotate(u'流通股\n' + str(chrate),
|
|||
|
|
# fontproperties=__font_properties__,
|
|||
|
|
# xy=(idx, yhighlim/1.1), xycoords='data',
|
|||
|
|
# xytext=(10, 5), textcoords='offset points', size=5, verticalalignment="center",
|
|||
|
|
# bbox=dict(boxstyle="round", facecolor='white', edgecolor='yellow'),
|
|||
|
|
# arrowprops=dict(arrowstyle="wedge,tail_width=1.",
|
|||
|
|
# facecolor='white', edgecolor='yellow',
|
|||
|
|
# patchA=None,
|
|||
|
|
# patchB=el,
|
|||
|
|
# relpos=(0.2, 0.8),
|
|||
|
|
# connectionstyle="arc3,rad=-0.1"),
|
|||
|
|
# alpha=0.5
|
|||
|
|
# )
|
|||
|
|
|
|||
|
|
if self._name == 'pricefs':
|
|||
|
|
fsize= 3.0
|
|||
|
|
ycoord= yhighlim/1.1
|
|||
|
|
alpha= 1.0
|
|||
|
|
else:
|
|||
|
|
fsize= 5.0
|
|||
|
|
ycoord= yhighlim/1.05
|
|||
|
|
alpha= 0.8
|
|||
|
|
|
|||
|
|
for record in changerecs:
|
|||
|
|
datestr= record[u'变更日期']
|
|||
|
|
chrate= record[u'变更比']
|
|||
|
|
dstr= [ds for ds in self._dates if ds >= datestr][0]
|
|||
|
|
idx= self._dates.index(dstr)
|
|||
|
|
axes.plot([idx, idx], [ylowlim, yhighlim], '-', color='yellow', linewidth=0.1)
|
|||
|
|
label= axes.text( \
|
|||
|
|
idx, ycoord, \
|
|||
|
|
u'流通股 ' + str(chrate) + u'\n' + datestr, \
|
|||
|
|
fontproperties=__font_properties__, \
|
|||
|
|
horizontalalignment='left', \
|
|||
|
|
verticalalignment='top', \
|
|||
|
|
color='yellow', \
|
|||
|
|
alpha=alpha
|
|||
|
|
)
|
|||
|
|
label.set_fontsize(fsize)
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
def plot_candlestick_2(self):
|
|||
|
|
'''
|
|||
|
|
绘制第二条 K 线
|
|||
|
|
'''
|
|||
|
|
xindex= self._xindex
|
|||
|
|
|
|||
|
|
axes= self._Axes_2
|
|||
|
|
|
|||
|
|
up= self._up_2
|
|||
|
|
down= self._down_2
|
|||
|
|
side= self._side_2
|
|||
|
|
|
|||
|
|
# 对开收盘价进行视觉修正
|
|||
|
|
for idx, poc in enumerate( self._zipoc_2 ):
|
|||
|
|
if poc[0] == poc[1] and None not in poc:
|
|||
|
|
self._open_2[idx]= poc[0] - 5 # 稍微偏离一点，使得在图线上不致于完全看不到
|
|||
|
|
self._close_2[idx]= poc[1] + 5
|
|||
|
|
|
|||
|
|
rarray_open= numpy.array(self._open_2)
|
|||
|
|
rarray_close= numpy.array(self._close_2)
|
|||
|
|
rarray_high= numpy.array(self._high_2)
|
|||
|
|
rarray_low= numpy.array(self._low_2)
|
|||
|
|
|
|||
|
|
# XXX: 如果 up, down, side 里有一个全部为 False 组成，那么 vlines() 会报错。
|
|||
|
|
# XXX: 可以使用 alpha 参数调节透明度
|
|||
|
|
if True in up:
|
|||
|
|
axes.vlines(xindex[up], rarray_low[up], rarray_high[up], edgecolor='0.7', linewidth=0.6, label='_nolegend_', alpha=0.5)
|
|||
|
|
axes.vlines(xindex[up], rarray_open[up], rarray_close[up], edgecolor='0.7', linewidth=3.0, label='_nolegend_', alpha=0.5)
|
|||
|
|
|
|||
|
|
if True in down:
|
|||
|
|
axes.vlines(xindex[down], rarray_low[down], rarray_high[down], edgecolor='0.3', linewidth=0.6, label='_nolegend_', alpha=0.5)
|
|||
|
|
axes.vlines(xindex[down], rarray_open[down], rarray_close[down], edgecolor='0.3', linewidth=3.0, label='_nolegend_', alpha=0.5)
|
|||
|
|
|
|||
|
|
if True in side:
|
|||
|
|
axes.vlines(xindex[side], rarray_low[side], rarray_high[side], edgecolor='1.0', linewidth=0.6, label='_nolegend_', alpha=1.0)
|
|||
|
|
axes.vlines(xindex[side], rarray_open[side], rarray_close[side], edgecolor='1.0', linewidth=3.0, label='_nolegend_', alpha=1.0)
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
def plot_capitalinfo(self):
|
|||
|
|
'''
|
|||
|
|
绘制行情首日和尾日的股本信息
|
|||
|
|
'''
|
|||
|
|
def find_biggestblank(didx):
|
|||
|
|
'''
|
|||
|
|
找出 X 轴某个位置图中最大的空隙
|
|||
|
|
'''
|
|||
|
|
pstart= self._open[0]
|
|||
|
|
ptarget= self._open[didx]
|
|||
|
|
|
|||
|
|
compseq= [yhighlim, ptarget, ylowlim]
|
|||
|
|
|
|||
|
|
if u'开盘二' in quotes:
|
|||
|
|
pstart_2= self._open_2[0]
|
|||
|
|
ptarget_2= self._open_2[didx]
|
|||
|
|
padjust= int(ptarget_2 * pstart / float(pstart_2))
|
|||
|
|
compseq.append(padjust)
|
|||
|
|
|
|||
|
|
compseq.sort(reverse=True) # 图中的三个或四个关键点，高到底排序
|
|||
|
|
|
|||
|
|
diff, hi, low= max([(math.log(compseq[idx]/float(compseq[idx+1]), expbase), compseq[idx], compseq[idx+1]) for idx in range(len(compseq)-1)])
|
|||
|
|
|
|||
|
|
# XXX: for debugging
|
|||
|
|
# print(compseq)
|
|||
|
|
# print([diff, hi, low])
|
|||
|
|
|
|||
|
|
return (hi*low)**0.5 # 相乘再开平方，在 log 坐标系里看起来就是在中间位置。
|
|||
|
|
|
|||
|
|
def repr_int(intnum):
|
|||
|
|
'''
|
|||
|
|
123456789 --> '1,2345,6789'
|
|||
|
|
'''
|
|||
|
|
if type(intnum) not in (int, long): return str(intnum)
|
|||
|
|
|
|||
|
|
if intnum == 0: return '0'
|
|||
|
|
|
|||
|
|
if intnum < 0:
|
|||
|
|
intnum= -intnum
|
|||
|
|
isminus= True
|
|||
|
|
else:
|
|||
|
|
isminus= False
|
|||
|
|
|
|||
|
|
intstr= str(intnum)
|
|||
|
|
intstr= '0'*((4-len(intstr)%4)%4) + intstr
|
|||
|
|
intlist= [intstr[i:i+4] for i in range(0, len(intstr), 4)]
|
|||
|
|
|
|||
|
|
intlist[0]= intlist[0].lstrip('0')
|
|||
|
|
|
|||
|
|
return ('-' + ','.join(intlist)) if isminus else ','.join(intlist)
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
pdata= self._pdata
|
|||
|
|
quotes= pdata[u'行情']
|
|||
|
|
|
|||
|
|
ylowlim= self._ylowlim
|
|||
|
|
yhighlim= self._yhighlim
|
|||
|
|
length= self._length
|
|||
|
|
expbase= self._expbase
|
|||
|
|
capinfo= pdata[u'公司信息'][u'股本变更记录']
|
|||
|
|
axes= self._Axes
|
|||
|
|
firstday, lastday= self._dates[0], self._dates[-1]
|
|||
|
|
|
|||
|
|
fsize= 5.0 if self._name == 'price' else 3.0
|
|||
|
|
|
|||
|
|
# 首日总股本与流通股信息
|
|||
|
|
#====================================================================================================================================
|
|||
|
|
chunk= [rec for rec in capinfo if rec[u'变更日期'] <= firstday]
|
|||
|
|
|
|||
|
|
if chunk:
|
|||
|
|
allshares= repr_int(chunk[-1][u'总股本'])
|
|||
|
|
circulating= repr_int(chunk[-1][u'流通股'])
|
|||
|
|
else:
|
|||
|
|
allshares= 'None'
|
|||
|
|
circulating= 'None'
|
|||
|
|
|
|||
|
|
infostr= u'总股本: ' + allshares + '\n' + u'流通股: ' + circulating
|
|||
|
|
|
|||
|
|
label= axes.text(0, find_biggestblank(didx=0), infostr, fontproperties=__font_properties__, color='0.7')
|
|||
|
|
label.set_fontsize(fsize)
|
|||
|
|
# label.set_zorder(0) # XXX: 放在底层
|
|||
|
|
|
|||
|
|
# 尾日总股本与流通股信息
|
|||
|
|
#====================================================================================================================================
|
|||
|
|
chunk= [rec for rec in capinfo if rec[u'变更日期'] <= lastday]
|
|||
|
|
if chunk:
|
|||
|
|
allshares= repr_int(chunk[-1][u'总股本'])
|
|||
|
|
circulating= repr_int(chunk[-1][u'流通股'])
|
|||
|
|
else:
|
|||
|
|
allshares= 'None'
|
|||
|
|
circulating= 'None'
|
|||
|
|
|
|||
|
|
infostr= u'总股本: ' + allshares + '\n' + u'流通股: ' + circulating
|
|||
|
|
|
|||
|
|
label= axes.text(length-1, find_biggestblank(didx=length-1), infostr, fontproperties=__font_properties__, horizontalalignment='right', color='0.7')
|
|||
|
|
label.set_fontsize(fsize)
|
|||
|
|
# label.set_zorder(0) # XXX: 放在底层
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
def plot_usernotes(self):
|
|||
|
|
'''
|
|||
|
|
|
|||
|
|
'''
|
|||
|
|
pdata= self._pdata
|
|||
|
|
quotes= pdata[u'行情']
|
|||
|
|
sidx= self._pdata[u'任务描述'][u'起始偏移']
|
|||
|
|
eidx= self._pdata[u'任务描述'][u'结束偏移']
|
|||
|
|
|
|||
|
|
axes= self._Axes
|
|||
|
|
usernotes= pdata[u'用户标记']
|
|||
|
|
|
|||
|
|
alldates= quotes[u'日期'][sidx:eidx]
|
|||
|
|
lowprices= quotes[u'最低'][sidx:eidx]
|
|||
|
|
expbase= self._expbase
|
|||
|
|
|
|||
|
|
# 绘制短线买点标记
|
|||
|
|
for note in usernotes:
|
|||
|
|
if note[u'类型'] == u'筛选结果':
|
|||
|
|
dstr= note[u'日期']
|
|||
|
|
|
|||
|
|
# 日期不在绘图区间范围内，忽略
|
|||
|
|
if dstr not in alldates:
|
|||
|
|
continue
|
|||
|
|
|
|||
|
|
# 决定箭头的颜色
|
|||
|
|
result= note[u'结果']
|
|||
|
|
color= 'magenta' if result == u'上涨' else 'cyan' if result == u'下跌' else 'white'
|
|||
|
|
|
|||
|
|
# 箭头的起始位置
|
|||
|
|
idx= alldates.index(dstr)
|
|||
|
|
xpos= idx
|
|||
|
|
ypos= lowprices[idx] / expbase
|
|||
|
|
|
|||
|
|
# 箭头的长度
|
|||
|
|
dx= 0.0
|
|||
|
|
dy= ypos * (expbase-1) * 0.9
|
|||
|
|
|
|||
|
|
# 头端的长度
|
|||
|
|
head_length= dy * 0.2
|
|||
|
|
|
|||
|
|
# 绘制箭头
|
|||
|
|
arrow_params={'length_includes_head':True, 'shape':'full', 'head_starts_at_zero':False}
|
|||
|
|
axes.arrow(xpos, ypos, dx, dy, facecolor=color, edgecolor=color, linewidth=0.7, head_width=0.6, head_length=head_length, **arrow_params)
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
def plot_vlines(self):
|
|||
|
|
|
|||
|
|
xindex= self._xindex
|
|||
|
|
|
|||
|
|
up= self._up
|
|||
|
|
down= self._down
|
|||
|
|
side= self._side
|
|||
|
|
|
|||
|
|
axes= self._Axes
|
|||
|
|
|
|||
|
|
lwidth= 2.4 * self._shrink
|
|||
|
|
|
|||
|
|
# 绘制 K 线部分
|
|||
|
|
#==================================================================================================================================================
|
|||
|
|
rarray_high= numpy.array(self._high)
|
|||
|
|
rarray_low= numpy.array(self._low)
|
|||
|
|
|
|||
|
|
# XXX: 如果 up, down, side 里有一个全部为 False 组成，那么 vlines() 会报错。
|
|||
|
|
# XXX: 可以使用 alpha 参数调节透明度
|
|||
|
|
if True in up:
|
|||
|
|
axes.vlines(xindex[up], rarray_low[up], rarray_high[up], edgecolor='red', linewidth=lwidth, label='_nolegend_', alpha=0.5)
|
|||
|
|
|
|||
|
|
if True in down:
|
|||
|
|
axes.vlines(xindex[down], rarray_low[down], rarray_high[down], edgecolor='green', linewidth=lwidth, label='_nolegend_', alpha=0.5)
|
|||
|
|
|
|||
|
|
if True in side:
|
|||
|
|
axes.vlines(xindex[side], rarray_low[side], rarray_high[side], edgecolor='0.7', linewidth=lwidth, label='_nolegend_', alpha=0.5)
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
def plot_vlines_2(self):
|
|||
|
|
xindex= self._xindex
|
|||
|
|
|
|||
|
|
axes= self._Axes_2
|
|||
|
|
|
|||
|
|
up= self._up_2
|
|||
|
|
down= self._down_2
|
|||
|
|
side= self._side_2
|
|||
|
|
|
|||
|
|
lwidth= 2.4 * self._shrink
|
|||
|
|
|
|||
|
|
rarray_high= numpy.array(self._high_2)
|
|||
|
|
rarray_low= numpy.array(self._low_2)
|
|||
|
|
|
|||
|
|
# XXX: 如果 up, down, side 里有一个全部为 False 组成，那么 vlines() 会报错。
|
|||
|
|
# XXX: 可以使用 alpha 参数调节透明度
|
|||
|
|
if True in up:
|
|||
|
|
axes.vlines(xindex[up], rarray_low[up], rarray_high[up], edgecolor='0.7', linewidth=lwidth, label='_nolegend_', alpha=0.5)
|
|||
|
|
|
|||
|
|
if True in down:
|
|||
|
|
axes.vlines(xindex[down], rarray_low[down], rarray_high[down], edgecolor='0.3', linewidth=lwidth, label='_nolegend_', alpha=0.5)
|
|||
|
|
|
|||
|
|
if True in side:
|
|||
|
|
axes.vlines(xindex[side], rarray_low[side], rarray_high[side], edgecolor='1.0', linewidth=lwidth, label='_nolegend_', alpha=1.0)
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
def plot_datenotes(self):
|
|||
|
|
'''
|
|||
|
|
内部的日期注释，由派生类定义
|
|||
|
|
'''
|
|||
|
|
pass
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
def plot_pricenotes(self):
|
|||
|
|
'''
|
|||
|
|
内部的价格注释，由派生类定义
|
|||
|
|
'''
|
|||
|
|
pass
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
class SubPlot_PriceFullSpan(SubPlot_PriceBase):
|
|||
|
|
'''
|
|||
|
|
|
|||
|
|
'''
|
|||
|
|
|
|||
|
|
def plot(self):
|
|||
|
|
'''
|
|||
|
|
绘图
|
|||
|
|
'''
|
|||
|
|
pdata= self._pdata
|
|||
|
|
# if u'简化' in pdata:
|
|||
|
|
# self.plot_simplified()
|
|||
|
|
# else:
|
|||
|
|
# self.plot_candlestick()
|
|||
|
|
|
|||
|
|
self.plot_vlines()
|
|||
|
|
self.plot_average()
|
|||
|
|
|
|||
|
|
if u'相对复权' in pdata[u'行情']:
|
|||
|
|
self.plot_adjustnotes()
|
|||
|
|
|
|||
|
|
if u'流通股变更' in pdata[u'公司信息']:
|
|||
|
|
self.plot_capchangenotes()
|
|||
|
|
|
|||
|
|
if u'股本变更记录' in pdata[u'公司信息']:
|
|||
|
|
self.plot_capitalinfo()
|
|||
|
|
|
|||
|
|
self.set_xticks()
|
|||
|
|
self.set_yticks()
|
|||
|
|
|
|||
|
|
if u'开盘二' in pdata[u'行情']:
|
|||
|
|
self.plot_vlines_2()
|
|||
|
|
self.set_xticks_2()
|
|||
|
|
self.set_yticks_2()
|
|||
|
|
|
|||
|
|
self.plot_datenotes()
|
|||
|
|
|
|||
|
|
if 'price' in self._parent._subplots:
|
|||
|
|
self.plot_windowspan()
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
def plot_datenotes(self):
|
|||
|
|
'''
|
|||
|
|
日期在图中间的显示
|
|||
|
|
'''
|
|||
|
|
ylowlim= self._ylowlim
|
|||
|
|
|
|||
|
|
axes= self._Axes
|
|||
|
|
|
|||
|
|
sdindex= self._xparams['sdindex']
|
|||
|
|
mdindex= self._xparams['mdindex']
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
# 每季度第一个交易日
|
|||
|
|
for ix in sdindex:
|
|||
|
|
newlab= axes.text(ix - (1/self._shrink), ylowlim*1.03, self._dates[ix])
|
|||
|
|
newlab.set_font_properties(__font_properties__)
|
|||
|
|
newlab.set_color('0.3')
|
|||
|
|
newlab.set_fontsize(4)
|
|||
|
|
newlab.set_rotation('45')
|
|||
|
|
# newlab.set_rotation('vertical')
|
|||
|
|
# newlab.set_horizontalalignment('left')
|
|||
|
|
# newlab.set_verticalalignment('bottom')
|
|||
|
|
# newlab.set_verticalalignment('center')
|
|||
|
|
newlab.set_zorder(0) # XXX: 放在底层
|
|||
|
|
|
|||
|
|
|
|||
|
|
# 每月第一个交易日
|
|||
|
|
for ix in mdindex:
|
|||
|
|
newlab= axes.text(ix - (0.8/self._shrink), ylowlim * 1.03, self._dates[ix])
|
|||
|
|
newlab.set_font_properties(__font_properties__)
|
|||
|
|
newlab.set_color('0.3')
|
|||
|
|
newlab.set_fontsize(3)
|
|||
|
|
newlab.set_rotation('45')
|
|||
|
|
# newlab.set_rotation('vertical')
|
|||
|
|
# newlab.set_horizontalalignment('left')
|
|||
|
|
# newlab.set_verticalalignment('top') # 不行
|
|||
|
|
# newlab.set_verticalalignment('center')
|
|||
|
|
# newlab.set_verticalalignment('bottom')
|
|||
|
|
newlab.set_zorder(0) # XXX: 放在底层
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
def plot_windowspan(self):
|
|||
|
|
|
|||
|
|
axes= self._Axes
|
|||
|
|
jobstat= self._pdata[u'任务描述']
|
|||
|
|
sindex, eindex= jobstat[u'起始偏移'], jobstat[u'结束偏移']
|
|||
|
|
hibdry, lobdry= self._parent._subplots['price'].get_ylimits()
|
|||
|
|
|
|||
|
|
xcoord= sindex - 1
|
|||
|
|
width= eindex - sindex + 1
|
|||
|
|
ycoord= lobdry
|
|||
|
|
height= hibdry - lobdry
|
|||
|
|
window= matplotlib.patches.Rectangle((xcoord, ycoord), width, height, fill=False, edgecolor=__color_lightblue__, linewidth=0.3, alpha=0.7)
|
|||
|
|
window.set_zorder(-1) # 放在底层
|
|||
|
|
axes.add_patch(window)
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
class SubPlot_Price(SubPlot_PriceBase):
|
|||
|
|
'''
|
|||
|
|
|
|||
|
|
'''
|
|||
|
|
|
|||
|
|
def plot(self):
|
|||
|
|
'''
|
|||
|
|
绘图
|
|||
|
|
'''
|
|||
|
|
pdata= self._pdata
|
|||
|
|
# if u'简化' in pdata:
|
|||
|
|
# self.plot_simplified()
|
|||
|
|
# else:
|
|||
|
|
# self.plot_candlestick()
|
|||
|
|
|
|||
|
|
self.plot_candlestick()
|
|||
|
|
self.plot_average()
|
|||
|
|
|
|||
|
|
if u'相对复权' in pdata[u'行情']:
|
|||
|
|
self.plot_adjustnotes()
|
|||
|
|
|
|||
|
|
if u'流通股变更' in pdata[u'公司信息']:
|
|||
|
|
self.plot_capchangenotes()
|
|||
|
|
|
|||
|
|
if u'股本变更记录' in pdata[u'公司信息']:
|
|||
|
|
self.plot_capitalinfo()
|
|||
|
|
|
|||
|
|
if u'用户标记' in pdata:
|
|||
|
|
self.plot_usernotes()
|
|||
|
|
|
|||
|
|
self.set_xticks()
|
|||
|
|
self.set_yticks()
|
|||
|
|
|
|||
|
|
if u'开盘二' in pdata[u'行情']:
|
|||
|
|
self.plot_candlestick_2()
|
|||
|
|
self.set_xticks_2()
|
|||
|
|
self.set_yticks_2()
|
|||
|
|
|
|||
|
|
self.plot_pricenotes()
|
|||
|
|
self.plot_datenotes()
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
def plot_datenotes(self):
|
|||
|
|
'''
|
|||
|
|
日期在图中间的显示
|
|||
|
|
'''
|
|||
|
|
|
|||
|
|
ylowlim= self._ylowlim
|
|||
|
|
yhighlim= self._yhighlim
|
|||
|
|
|
|||
|
|
axes= self._Axes
|
|||
|
|
|
|||
|
|
mdindex= self._xparams['mdindex']
|
|||
|
|
wdindex= self._xparams['wdindex']
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
# 每月第一个交易日
|
|||
|
|
for iy in [ylowlim*1.1, yhighlim/1.21]:
|
|||
|
|
for ix in mdindex:
|
|||
|
|
newlab= axes.text(ix-1, iy, self._dates[ix])
|
|||
|
|
newlab.set_font_properties(__font_properties__)
|
|||
|
|
newlab.set_color('0.3')
|
|||
|
|
newlab.set_fontsize(4)
|
|||
|
|
newlab.set_rotation('vertical')
|
|||
|
|
# newlab.set_horizontalalignment('left')
|
|||
|
|
# newlab.set_verticalalignment('bottom')
|
|||
|
|
# newlab.set_verticalalignment('center')
|
|||
|
|
newlab.set_zorder(0) # XXX: 放在底层
|
|||
|
|
|
|||
|
|
|
|||
|
|
# 每周第一个交易日，根据这个可以推算出全部确切的日期。
|
|||
|
|
# for iy in minorticks[0:-1:6]:
|
|||
|
|
for iy in [ylowlim*1.01, yhighlim/1.09]:
|
|||
|
|
for ix in wdindex:
|
|||
|
|
newlab= axes.text(ix-0.8, iy, self._dates[ix])
|
|||
|
|
newlab.set_font_properties(__font_properties__)
|
|||
|
|
newlab.set_color('0.3')
|
|||
|
|
newlab.set_fontsize(3)
|
|||
|
|
newlab.set_rotation('vertical')
|
|||
|
|
# newlab.set_horizontalalignment('left')
|
|||
|
|
# newlab.set_verticalalignment('top') # 不行
|
|||
|
|
# newlab.set_verticalalignment('center')
|
|||
|
|
# newlab.set_verticalalignment('bottom')
|
|||
|
|
newlab.set_zorder(0) # XXX: 放在底层
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
def plot_pricenotes(self):
|
|||
|
|
# 价格数值在图中间的显示
|
|||
|
|
#==================================================================================================================================================
|
|||
|
|
|
|||
|
|
quotes= self._pdata[u'行情']
|
|||
|
|
|
|||
|
|
axes= self._Axes
|
|||
|
|
majorticks= self._ytickset['major']
|
|||
|
|
minorticks= self._ytickset['minor']
|
|||
|
|
|
|||
|
|
mdindex= self._xparams['mdindex']
|
|||
|
|
|
|||
|
|
def price_note(num):
|
|||
|
|
return str(round(num/1000.0, 2))
|
|||
|
|
|
|||
|
|
if u'开盘二' in quotes:
|
|||
|
|
majorticks_2= self._ytickset['major_2']
|
|||
|
|
minorticks_2= self._ytickset['minor_2']
|
|||
|
|
|
|||
|
|
for iy, iy2 in zip(sorted(majorticks[:-1] + minorticks[1:-1]), sorted(majorticks_2[:-1] + minorticks_2[1:-1])):
|
|||
|
|
for ix in mdindex[1:-1:3]:
|
|||
|
|
newlab= axes.text(ix+6, iy*1.001, price_note(iy) + ' / ' + price_note(iy2))
|
|||
|
|
newlab.set_font_properties(__font_properties__)
|
|||
|
|
newlab.set_color('0.3')
|
|||
|
|
newlab.set_fontsize(3)
|
|||
|
|
newlab.set_zorder(0) # XXX: 放在底层
|
|||
|
|
else:
|
|||
|
|
for iy in sorted(majorticks[:-1] + minorticks[1:-1]):
|
|||
|
|
for ix in mdindex[1:-1:3]:
|
|||
|
|
newlab= axes.text(ix+9, iy*1.001, price_note(iy))
|
|||
|
|
newlab.set_font_properties(__font_properties__)
|
|||
|
|
newlab.set_color('0.3')
|
|||
|
|
newlab.set_fontsize(3)
|
|||
|
|
newlab.set_zorder(0) # XXX: 放在底层
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
class SubPlot_TORateBase:
|
|||
|
|
'''
|
|||
|
|
换手率子图
|
|||
|
|
'''
|
|||
|
|
|
|||
|
|
def __init__(self, pdata, parent, xparams, name):
|
|||
|
|
self._name= name
|
|||
|
|
self._pdata= pdata
|
|||
|
|
self._parent= parent
|
|||
|
|
self._xparams= xparams
|
|||
|
|
self._shrink= __shrink__ if name == 'toratefs' else 1.0
|
|||
|
|
|
|||
|
|
self._tostep= 0 # 每一格代表的换手率数值
|
|||
|
|
|
|||
|
|
self._yrange= 0
|
|||
|
|
|
|||
|
|
self._xsize= 0 # int
|
|||
|
|
self._ysize= 0 # int
|
|||
|
|
|
|||
|
|
self._Axes= None
|
|||
|
|
self._AxisX= None
|
|||
|
|
self._AxisY= None
|
|||
|
|
|
|||
|
|
if u'换手率二' in pdata[u'行情']:
|
|||
|
|
self._Axes_2= None
|
|||
|
|
self._AxisX_2= None
|
|||
|
|
self._AxisY_2= None
|
|||
|
|
self._tostep_2= 0
|
|||
|
|
|
|||
|
|
# 绘图数据
|
|||
|
|
quotes= pdata[u'行情']
|
|||
|
|
|
|||
|
|
if name == 'toratefs':
|
|||
|
|
self._dates= quotes[u'日期']
|
|||
|
|
self._open= quotes[u'开盘']
|
|||
|
|
self._close= quotes[u'收盘']
|
|||
|
|
self._high= quotes[u'最高']
|
|||
|
|
self._low= quotes[u'最低']
|
|||
|
|
if u'简化' in quotes: self._simple= quotes[u'简化']
|
|||
|
|
|
|||
|
|
if u'换手率' in quotes: self._torate= quotes[u'换手率']
|
|||
|
|
if u'成交量' in quotes: self._volume= quotes[u'成交量']
|
|||
|
|
if u'成交额' in quotes: self._turnover= quotes[u'成交额']
|
|||
|
|
|
|||
|
|
if u'开盘二' in quotes:
|
|||
|
|
self._open_2= quotes[u'开盘二']
|
|||
|
|
self._close_2= quotes[u'收盘二']
|
|||
|
|
self._high_2= quotes[u'最高二']
|
|||
|
|
self._low_2= quotes[u'最低二']
|
|||
|
|
if u'简化二' in quotes: self._simple_2= quotes[u'简化二']
|
|||
|
|
|
|||
|
|
if u'换手率二' in quotes: self._torate_2= quotes[u'换手率二']
|
|||
|
|
if u'成交量二' in quotes: self._volume_2= quotes[u'成交量二']
|
|||
|
|
if u'成交额二' in quotes: self._turnover_2= quotes[u'成交额二']
|
|||
|
|
|
|||
|
|
else:
|
|||
|
|
sidx, eidx= pdata[u'任务描述'][u'起始偏移'], pdata[u'任务描述'][u'结束偏移']
|
|||
|
|
|
|||
|
|
self._dates= quotes[u'日期'][sidx:eidx]
|
|||
|
|
self._open= quotes[u'开盘'][sidx:eidx]
|
|||
|
|
self._close= quotes[u'收盘'][sidx:eidx]
|
|||
|
|
self._high= quotes[u'最高'][sidx:eidx]
|
|||
|
|
self._low= quotes[u'最低'][sidx:eidx]
|
|||
|
|
if u'简化' in quotes: self._simple= quotes[u'简化'][sidx:eidx]
|
|||
|
|
|
|||
|
|
if u'换手率' in quotes: self._torate= quotes[u'换手率'][sidx:eidx]
|
|||
|
|
if u'成交量' in quotes: self._volume= quotes[u'成交量'][sidx:eidx]
|
|||
|
|
if u'成交额' in quotes: self._turnover= quotes[u'成交额'][sidx:eidx]
|
|||
|
|
|
|||
|
|
if u'开盘二' in quotes:
|
|||
|
|
self._open_2= quotes[u'开盘二'][sidx:eidx]
|
|||
|
|
self._close_2= quotes[u'收盘二'][sidx:eidx]
|
|||
|
|
self._high_2= quotes[u'最高二'][sidx:eidx]
|
|||
|
|
self._low_2= quotes[u'最低二'][sidx:eidx]
|
|||
|
|
if u'简化二' in quotes: self._simple_2= quotes[u'简化二'][sidx:eidx]
|
|||
|
|
|
|||
|
|
if u'换手率二' in quotes: self._torate_2= quotes[u'换手率二'][sidx:eidx]
|
|||
|
|
if u'成交量二' in quotes: self._volume_2= quotes[u'成交量二'][sidx:eidx]
|
|||
|
|
if u'成交额二' in quotes: self._turnover_2= quotes[u'成交额二'][sidx:eidx]
|
|||
|
|
|
|||
|
|
|
|||
|
|
# 衍生数据
|
|||
|
|
#==============================================================================================================
|
|||
|
|
self._length= len(self._dates)
|
|||
|
|
self._xindex= numpy.arange(self._length) # X 轴上的 index，一个辅助数据
|
|||
|
|
|
|||
|
|
self._zipoc= zip(self._open, self._close)
|
|||
|
|
self._up= numpy.array( [ True if po < pc and po is not None else False for po, pc in self._zipoc] ) # 标示出该天股价日内上涨的一个序列
|
|||
|
|
self._down= numpy.array( [ True if po > pc and po is not None else False for po, pc in self._zipoc] ) # 标示出该天股价日内下跌的一个序列
|
|||
|
|
self._side= numpy.array( [ True if po == pc and po is not None else False for po, pc in self._zipoc] ) # 标示出该天股价日内走平的一个序列
|
|||
|
|
|
|||
|
|
if u'开盘二' in quotes:
|
|||
|
|
self._zipoc_2= zip(self._open_2, self._close_2)
|
|||
|
|
self._up_2= numpy.array( [ True if po < pc and po is not None else False for po, pc in self._zipoc_2] ) # 标示出该天股价日内上涨的一个序列
|
|||
|
|
self._down_2= numpy.array( [ True if po > pc and po is not None else False for po, pc in self._zipoc_2] ) # 标示出该天股价日内下跌的一个序列
|
|||
|
|
self._side_2= numpy.array( [ True if po == pc and po is not None else False for po, pc in self._zipoc_2] ) # 标示出该天股价日内走平的一个序列
|
|||
|
|
|
|||
|
|
self._compute_size()
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
def _compute_size(self):
|
|||
|
|
'''
|
|||
|
|
根据 pdata 计算自身尺寸
|
|||
|
|
'''
|
|||
|
|
def _compute_step(maxto):
|
|||
|
|
'''
|
|||
|
|
maxto 是 换手率 最大值。返回每格单位（最小 500, 代表 0.5%）以及格数
|
|||
|
|
'''
|
|||
|
|
for i in range(9):
|
|||
|
|
if maxto > (4 * 500 * (2**i)): # 换手率最大是 100000, 代表 100%
|
|||
|
|
continue
|
|||
|
|
else:
|
|||
|
|
tostep= 500 * (2**i)
|
|||
|
|
tosize= int(round((maxto + tostep/2.0 - 1) / float(tostep), 0))
|
|||
|
|
break
|
|||
|
|
return (tostep, tosize)
|
|||
|
|
|
|||
|
|
quotes= self._pdata[u'行情']
|
|||
|
|
xmargin= self._xparams['xmargin']
|
|||
|
|
|
|||
|
|
self._xsize= (self._length + xmargin*2) * self._shrink
|
|||
|
|
|
|||
|
|
maxto= max(self._torate)
|
|||
|
|
self._tostep, self._yrange= _compute_step(maxto=maxto)
|
|||
|
|
|
|||
|
|
if u'换手率二' in quotes:
|
|||
|
|
maxto_2= max(self._torate_2)
|
|||
|
|
self._tostep_2, yrange_2= _compute_step(maxto=maxto_2)
|
|||
|
|
self._yrange= max(self._yrange, yrange_2) # 成交量部分在 Y 轴所占的 “份数”
|
|||
|
|
|
|||
|
|
self._ysize= self._yrange * self._shrink
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
def get_size(self):
|
|||
|
|
return (self._xsize, self._ysize)
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
def build_axes(self, figobj, rect):
|
|||
|
|
|
|||
|
|
# 第一只：添加 Axes 对象
|
|||
|
|
#==================================================================================================================================================
|
|||
|
|
axes= figobj.add_axes(rect, axis_bgcolor='black')
|
|||
|
|
|
|||
|
|
axes.set_axis_bgcolor('black')
|
|||
|
|
axes.set_axisbelow(True) # 网格线放在底层
|
|||
|
|
|
|||
|
|
# 第一只：改变坐标线的颜色
|
|||
|
|
#==================================================================================================================================================
|
|||
|
|
for child in axes.get_children():
|
|||
|
|
if isinstance(child, matplotlib.spines.Spine):
|
|||
|
|
child.set_color(__color_gold__)
|
|||
|
|
# child.set_zorder(3) # XXX: 放在上层，好像没什么用。
|
|||
|
|
|
|||
|
|
# 得到 X 轴 和 Y 轴 的两个 Axis 对象
|
|||
|
|
#==================================================================================================================================================
|
|||
|
|
xaxis= axes.get_xaxis()
|
|||
|
|
yaxis= axes.get_yaxis()
|
|||
|
|
|
|||
|
|
# 设置两个坐标轴上的 grid
|
|||
|
|
#==================================================================================================================================================
|
|||
|
|
xaxis.grid(True, 'major', color='0.3', linestyle='solid', linewidth=0.2)
|
|||
|
|
xaxis.grid(True, 'minor', color='0.3', linestyle='dotted', linewidth=0.1)
|
|||
|
|
|
|||
|
|
yaxis.grid(True, 'major', color='0.3', linestyle='solid', linewidth=0.2)
|
|||
|
|
yaxis.grid(True, 'minor', color='0.3', linestyle='solid', linewidth=0.1)
|
|||
|
|
|
|||
|
|
self._Axes= axes
|
|||
|
|
self._AxisX= xaxis
|
|||
|
|
self._AxisY= yaxis
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
if u'换手率二' in self._pdata[u'行情']:
|
|||
|
|
# 添加 Axes 对象
|
|||
|
|
#==================================================================================================================================================
|
|||
|
|
axes_2= axes.twinx()
|
|||
|
|
|
|||
|
|
# XXX: 下面这三行把第一个 axes 放在上面，这样不会被第二个 axes 的图形遮盖。用 zorder 不顶用。
|
|||
|
|
axes.figure.axes[-2:]= [axes_2, axes] # XXX: 把第一个 axes 放在上面，用 zorder 不顶用。
|
|||
|
|
axes.set_frame_on(False) # 如果不做此设定，axes_2 的内容会看不见
|
|||
|
|
axes_2.set_frame_on(True)
|
|||
|
|
|
|||
|
|
axes_2.set_axis_bgcolor('black')
|
|||
|
|
axes_2.set_axisbelow(True) # 网格线放在底层
|
|||
|
|
|
|||
|
|
# 改变坐标线的颜色
|
|||
|
|
#==================================================================================================================================================
|
|||
|
|
for child in axes_2.get_children():
|
|||
|
|
if isinstance(child, matplotlib.spines.Spine):
|
|||
|
|
child.set_color(__color_gold__)
|
|||
|
|
|
|||
|
|
# 得到 X 轴 和 Y 轴 的两个 Axis 对象
|
|||
|
|
#==================================================================================================================================================
|
|||
|
|
xaxis_2= axes_2.get_xaxis()
|
|||
|
|
yaxis_2= axes_2.get_yaxis()
|
|||
|
|
|
|||
|
|
# 设置网格线
|
|||
|
|
#==================================================================================================================================================
|
|||
|
|
# xaxis_2.grid(True, 'major', color='0.3', linestyle='solid', linewidth=0.2)
|
|||
|
|
# xaxis_2.grid(True, 'minor', color='0.3', linestyle='dotted', linewidth=0.1)
|
|||
|
|
|
|||
|
|
# yaxis_2.grid(True, 'major', color='0.3', linestyle='solid', linewidth=0.2)
|
|||
|
|
# yaxis_2.grid(True, 'minor', color='0.3', linestyle='dotted', linewidth=0.1)
|
|||
|
|
|
|||
|
|
self._Axes_2= axes_2
|
|||
|
|
self._AxisX_2= xaxis_2
|
|||
|
|
self._AxisY_2= yaxis_2
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
def get_axes(self):
|
|||
|
|
return self._Axes
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
def plot(self):
|
|||
|
|
'''
|
|||
|
|
绘制换手率图形
|
|||
|
|
'''
|
|||
|
|
self.plot_torate()
|
|||
|
|
self.set_xticks()
|
|||
|
|
self.set_yticks()
|
|||
|
|
|
|||
|
|
if u'换手率二' in self._pdata[u'行情']:
|
|||
|
|
self.plot_torate_2()
|
|||
|
|
self.set_xticks_2()
|
|||
|
|
self.set_yticks_2()
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
def plot_torate(self):
|
|||
|
|
'''
|
|||
|
|
绘制换手率
|
|||
|
|
'''
|
|||
|
|
|
|||
|
|
xindex= self._xindex
|
|||
|
|
stopset= self._xparams['mdindex'] if self._name == 'torate' else self._xparams['sdindex']
|
|||
|
|
axes= self._Axes
|
|||
|
|
|
|||
|
|
up= self._up
|
|||
|
|
down= self._down
|
|||
|
|
side= self._side
|
|||
|
|
|
|||
|
|
rarray_to= numpy.array(self._torate)
|
|||
|
|
tozeros= numpy.zeros(self._length) # 辅助数据
|
|||
|
|
|
|||
|
|
lwidth= 3.0 if self._name == 'torate' else 2.4 * self._shrink
|
|||
|
|
|
|||
|
|
# XXX: 如果 up/down/side 各项全部为 False，那么 vlines() 会报错。
|
|||
|
|
if True in up:
|
|||
|
|
axes.vlines(xindex[up], tozeros[up], rarray_to[up], edgecolor='red', linewidth=lwidth, label='_nolegend_', alpha=0.5)
|
|||
|
|
if True in down:
|
|||
|
|
axes.vlines(xindex[down], tozeros[down], rarray_to[down], edgecolor='green', linewidth=lwidth, label='_nolegend_', alpha=0.5)
|
|||
|
|
if True in side:
|
|||
|
|
axes.vlines(xindex[side], tozeros[side], rarray_to[side], edgecolor='0.7', linewidth=lwidth, label='_nolegend_', alpha=0.5)
|
|||
|
|
|
|||
|
|
# 绘制平均换手率（直线）
|
|||
|
|
toeffect= [num for num in self._torate if num is not None]
|
|||
|
|
toaverage= sum(toeffect) / float(len(toeffect))
|
|||
|
|
|
|||
|
|
axes.plot([-1, self._length], [toaverage, toaverage], '-', color='yellow', linewidth=0.2, alpha=0.7)
|
|||
|
|
|
|||
|
|
# 换手率数值在图中间的显示
|
|||
|
|
#==================================================================================================================================================
|
|||
|
|
for ix in stopset[2:-1:3]:
|
|||
|
|
newlab= axes.text(ix+8, toaverage, str(round(toaverage/1000.0, 2)) + '%')
|
|||
|
|
newlab.set_font_properties(__font_properties__)
|
|||
|
|
newlab.set_color('yellow')
|
|||
|
|
newlab.set_fontsize(3)
|
|||
|
|
# newlab.set_zorder(0) # XXX: 放在底层
|
|||
|
|
# newlab.set_verticalalignment('center')
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
def plot_torate_2(self):
|
|||
|
|
'''
|
|||
|
|
绘制第二条换手率柱状图
|
|||
|
|
'''
|
|||
|
|
quotes= self._pdata[u'行情']
|
|||
|
|
xindex= self._xindex
|
|||
|
|
axes= self._Axes_2
|
|||
|
|
|
|||
|
|
up= self._up_2
|
|||
|
|
down= self._down_2
|
|||
|
|
side= self._side_2
|
|||
|
|
|
|||
|
|
rarray_to= numpy.array(self._torate_2)
|
|||
|
|
tozeros= numpy.zeros(self._length) # 辅助数据
|
|||
|
|
|
|||
|
|
lwidth, alpha= (0.7, 0.5) if self._name == 'torate' else (0.3, 0.7)
|
|||
|
|
|
|||
|
|
# XXX: 如果 up/down/side 各项全部为 False，那么 vlines() 会报错。
|
|||
|
|
if True in up:
|
|||
|
|
axes.vlines(xindex[up], tozeros[up], rarray_to[up], edgecolor='0.7', linewidth=lwidth, label='_nolegend_', alpha=alpha)
|
|||
|
|
if True in down:
|
|||
|
|
axes.vlines(xindex[down], tozeros[down], rarray_to[down], edgecolor='0.3', linewidth=lwidth, label='_nolegend_', alpha=alpha)
|
|||
|
|
if True in side:
|
|||
|
|
axes.vlines(xindex[side], tozeros[side], rarray_to[side], edgecolor='0.7', linewidth=lwidth, label='_nolegend_', alpha=1.0)
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
def set_xticks(self):
|
|||
|
|
'''
|
|||
|
|
X 轴坐标
|
|||
|
|
'''
|
|||
|
|
length= self._length
|
|||
|
|
xmargin= self._xparams['xmargin']
|
|||
|
|
|
|||
|
|
axes= self._Axes
|
|||
|
|
xaxis= self._AxisX
|
|||
|
|
|
|||
|
|
# xaxis.set_tick_params(which='both', direction='out') # XXX: 坐标点设到外面去，也可以用 Axes.tick_params()，好像 matplotlib 1.0.1 才有
|
|||
|
|
|
|||
|
|
# 设定 X 轴坐标的范围
|
|||
|
|
#==================================================================================================================================================
|
|||
|
|
axes.set_xlim(-xmargin, length + xmargin)
|
|||
|
|
|
|||
|
|
xMajorLocator= self._xparams['xMajorLocator']
|
|||
|
|
xMinorLocator= self._xparams['xMinorLocator']
|
|||
|
|
xMajorFormatter= self._xparams['xMajorFormatter']
|
|||
|
|
xMinorFormatter= self._xparams['xMinorFormatter']
|
|||
|
|
|
|||
|
|
# 设定 X 轴的 Locator 和 Formatter
|
|||
|
|
xaxis.set_major_locator(xMajorLocator)
|
|||
|
|
xaxis.set_minor_locator(xMinorLocator)
|
|||
|
|
|
|||
|
|
if self._name == 'torate':
|
|||
|
|
xaxis.set_major_formatter(xMajorFormatter)
|
|||
|
|
xaxis.set_minor_formatter(xMinorFormatter)
|
|||
|
|
|
|||
|
|
# 设定 X 轴主要坐标点与辅助坐标点的样式
|
|||
|
|
for mal in axes.get_xticklabels(minor=False):
|
|||
|
|
mal.set_fontsize(4)
|
|||
|
|
mal.set_horizontalalignment('right')
|
|||
|
|
mal.set_rotation('45')
|
|||
|
|
|
|||
|
|
for mil in axes.get_xticklabels(minor=True):
|
|||
|
|
mil.set_fontsize(4)
|
|||
|
|
mil.set_color('blue')
|
|||
|
|
mil.set_horizontalalignment('right')
|
|||
|
|
mil.set_rotation('45')
|
|||
|
|
else:
|
|||
|
|
# 设为不可见
|
|||
|
|
for mal in axes.get_xticklabels(minor=False):
|
|||
|
|
mal.set_visible(False)
|
|||
|
|
|
|||
|
|
for mil in axes.get_xticklabels(minor=True):
|
|||
|
|
mil.set_visible(False)
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
def set_xticks_2(self):
|
|||
|
|
|
|||
|
|
length= self._length
|
|||
|
|
xmargin= self._xparams['xmargin']
|
|||
|
|
|
|||
|
|
axes= self._Axes_2
|
|||
|
|
xaxis= self._AxisX_2
|
|||
|
|
|
|||
|
|
# xaxis.set_tick_params(which='both', direction='out') # XXX: 坐标点设到外面去，也可以用 Axes.tick_params()，好像 matplotlib 1.0.1 才有
|
|||
|
|
|
|||
|
|
# 设定 X 轴坐标的范围
|
|||
|
|
#==================================================================================================================================================
|
|||
|
|
axes.set_xlim(-xmargin, length + xmargin)
|
|||
|
|
|
|||
|
|
xMajorLocator= self._xparams['xMajorLocator']
|
|||
|
|
xMinorLocator= self._xparams['xMinorLocator']
|
|||
|
|
|
|||
|
|
# 设定 X 轴的 Locator 和 Formatter
|
|||
|
|
xaxis.set_major_locator(xMajorLocator)
|
|||
|
|
xaxis.set_minor_locator(xMinorLocator)
|
|||
|
|
|
|||
|
|
# 设为不可见
|
|||
|
|
for mal in axes.get_xticklabels(minor=False):
|
|||
|
|
mal.set_visible(False)
|
|||
|
|
|
|||
|
|
for mil in axes.get_xticklabels(minor=True):
|
|||
|
|
mil.set_visible(False)
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
def set_yticks(self):
|
|||
|
|
'''
|
|||
|
|
设置 Y 轴坐标
|
|||
|
|
'''
|
|||
|
|
axes= self._Axes
|
|||
|
|
yaxis= self._AxisY
|
|||
|
|
tostep= self._tostep
|
|||
|
|
yrange= self._yrange
|
|||
|
|
stopset= self._xparams['mdindex'] if self._name == 'torate' else self._xparams['sdindex']
|
|||
|
|
|
|||
|
|
# 设定换手率 Y 轴坐标的范围
|
|||
|
|
#==================================================================================================================================================
|
|||
|
|
axes.set_ylim(0, tostep*yrange)
|
|||
|
|
|
|||
|
|
# 主要坐标点
|
|||
|
|
#==================================================================================================================================================
|
|||
|
|
majorticks= [tostep*i for i in range(yrange)]
|
|||
|
|
yMajorLocator= FixedLocator(numpy.array(majorticks))
|
|||
|
|
|
|||
|
|
# 确定 Y 轴的 MajorFormatter
|
|||
|
|
def y_major_formatter(num, pos=None):
|
|||
|
|
return str(round(num/1000.0, 2)) + '%'
|
|||
|
|
|
|||
|
|
yMajorFormatter= FuncFormatter(y_major_formatter)
|
|||
|
|
|
|||
|
|
# 确定 Y 轴的 MinorFormatter
|
|||
|
|
yMinorFormatter= NullFormatter()
|
|||
|
|
|
|||
|
|
# 第一只：设定 X 轴的 Locator 和 Formatter
|
|||
|
|
yaxis.set_major_locator(yMajorLocator)
|
|||
|
|
yaxis.set_major_formatter(yMajorFormatter)
|
|||
|
|
|
|||
|
|
# 设定 Y 轴主要坐标点的样式
|
|||
|
|
for mal in axes.get_yticklabels(minor=False):
|
|||
|
|
mal.set_font_properties(__font_properties__)
|
|||
|
|
mal.set_fontsize(5) # 这个必须放在前一句后面，否则作用会被覆盖
|
|||
|
|
|
|||
|
|
# 辅助坐标点
|
|||
|
|
#==================================================================================================================================================
|
|||
|
|
if self._name == 'torate':
|
|||
|
|
minorticks= list( itertools.chain.from_iterable( mi for mi in [[ma + (tostep/4.0)*i for i in range(1, 4)] for ma in majorticks] ) )
|
|||
|
|
yMinorLocator= FixedLocator(numpy.array(minorticks))
|
|||
|
|
yaxis.set_minor_locator(yMinorLocator)
|
|||
|
|
|
|||
|
|
def y_minor_formatter(num, pos=None):
|
|||
|
|
return str(round(num/1000.0, 3)) + '%'
|
|||
|
|
|
|||
|
|
yMinorFormatter= FuncFormatter(y_minor_formatter)
|
|||
|
|
|
|||
|
|
yaxis.set_minor_formatter(yMinorFormatter)
|
|||
|
|
|
|||
|
|
# 设定 Y 轴主要坐标点的样式
|
|||
|
|
for mil in axes.get_yticklabels(minor=True):
|
|||
|
|
mil.set_font_properties(__font_properties__)
|
|||
|
|
mil.set_fontsize(4) # 这个必须放在前一句后面，否则作用会被覆盖
|
|||
|
|
|
|||
|
|
else:
|
|||
|
|
|
|||
|
|
# minorticks= list( itertools.chain.from_iterable( mi for mi in [[ma + (tostep/4.0)*i for i in range(1, 4)] for ma in majorticks] ) )
|
|||
|
|
minorticks= list( [ma + (tostep/2.0) for ma in majorticks] )
|
|||
|
|
yMinorLocator= FixedLocator(numpy.array(minorticks))
|
|||
|
|
yaxis.set_minor_locator(yMinorLocator)
|
|||
|
|
|
|||
|
|
# 设定 Y 轴主要坐标点的样式
|
|||
|
|
for mil in axes.get_yticklabels(minor=True):
|
|||
|
|
mil.set_visible(False)
|
|||
|
|
|
|||
|
|
# 换手率数值在图中间的显示
|
|||
|
|
#==================================================================================================================================================
|
|||
|
|
for iy in range(int(tostep/2.0), tostep*yrange, int(tostep/2.0)):
|
|||
|
|
for ix in stopset[1:-1:3]:
|
|||
|
|
newlab= axes.text(ix+8, iy, y_major_formatter(iy))
|
|||
|
|
newlab.set_font_properties(__font_properties__)
|
|||
|
|
newlab.set_color('0.3')
|
|||
|
|
newlab.set_fontsize(3)
|
|||
|
|
newlab.set_zorder(0) # XXX: 放在底层
|
|||
|
|
# newlab.set_verticalalignment('center')
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
def set_yticks_2(self):
|
|||
|
|
'''
|
|||
|
|
设置 Y 轴坐标
|
|||
|
|
'''
|
|||
|
|
axes= self._Axes_2
|
|||
|
|
yaxis= self._AxisY_2
|
|||
|
|
tostep= self._tostep_2
|
|||
|
|
yrange= self._yrange # 与 1 是一样的
|
|||
|
|
|
|||
|
|
# 设定换手率 Y 轴坐标的范围
|
|||
|
|
#==================================================================================================================================================
|
|||
|
|
axes.set_ylim(0, tostep*yrange)
|
|||
|
|
|
|||
|
|
# 主要坐标点
|
|||
|
|
#==================================================================================================================================================
|
|||
|
|
majorticks= [tostep*i for i in range(yrange)]
|
|||
|
|
yMajorLocator= FixedLocator(numpy.array(majorticks))
|
|||
|
|
|
|||
|
|
# 确定 Y 轴的 MajorFormatter
|
|||
|
|
def y_major_formatter(num, pos=None):
|
|||
|
|
return str(round(num/1000.0, 2)) + '%'
|
|||
|
|
|
|||
|
|
yMajorFormatter= FuncFormatter(y_major_formatter)
|
|||
|
|
|
|||
|
|
# 确定 Y 轴的 MinorFormatter
|
|||
|
|
yMinorFormatter= NullFormatter()
|
|||
|
|
|
|||
|
|
# 第一只：设定 X 轴的 Locator 和 Formatter
|
|||
|
|
yaxis.set_major_locator(yMajorLocator)
|
|||
|
|
yaxis.set_major_formatter(yMajorFormatter)
|
|||
|
|
|
|||
|
|
# 设定 Y 轴主要坐标点的样式
|
|||
|
|
for mal in axes.get_yticklabels(minor=False):
|
|||
|
|
mal.set_font_properties(__font_properties__)
|
|||
|
|
mal.set_fontsize(5) # 这个必须放在前一句后面，否则作用会被覆盖
|
|||
|
|
|
|||
|
|
# 辅助坐标点
|
|||
|
|
#==================================================================================================================================================
|
|||
|
|
if self._name == 'torate':
|
|||
|
|
minorticks= list( itertools.chain.from_iterable( mi for mi in [[ma + (tostep/4.0)*i for i in range(1, 4)] for ma in majorticks] ) )
|
|||
|
|
yMinorLocator= FixedLocator(numpy.array(minorticks))
|
|||
|
|
|
|||
|
|
def y_minor_formatter(num, pos=None):
|
|||
|
|
return str(round(num/1000.0, 3)) + '%'
|
|||
|
|
|
|||
|
|
yMinorFormatter= FuncFormatter(y_minor_formatter)
|
|||
|
|
|
|||
|
|
yaxis.set_minor_locator(yMinorLocator)
|
|||
|
|
yaxis.set_minor_formatter(yMinorFormatter)
|
|||
|
|
|
|||
|
|
# 设定 Y 轴主要坐标点的样式
|
|||
|
|
for mil in axes.get_yticklabels(minor=True):
|
|||
|
|
mil.set_font_properties(__font_properties__)
|
|||
|
|
mil.set_fontsize(4) # 这个必须放在前一句后面，否则作用会被覆盖
|
|||
|
|
|
|||
|
|
else:
|
|||
|
|
minorticks= list( [ma + (tostep/2.0) for ma in majorticks] )
|
|||
|
|
yMinorLocator= FixedLocator(numpy.array(minorticks))
|
|||
|
|
|
|||
|
|
yaxis.set_minor_locator(yMinorLocator)
|
|||
|
|
|
|||
|
|
# 设定 Y 轴主要坐标点的样式
|
|||
|
|
for mil in axes.get_yticklabels(minor=True):
|
|||
|
|
mil.set_visible(False)
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
class SubPlot_TORate(SubPlot_TORateBase):
|
|||
|
|
pass
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
class SubPlot_TORateFullSpan(SubPlot_TORateBase):
|
|||
|
|
pass
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
class MyFigure:
|
|||
|
|
'''
|
|||
|
|
|
|||
|
|
'''
|
|||
|
|
def __init__(self, pdata):
|
|||
|
|
self._pdata= pdata # 绘图数据
|
|||
|
|
|
|||
|
|
self._figfacecolor= __color_pink__
|
|||
|
|
self._figedgecolor= __color_navy__
|
|||
|
|
self._figdpi= 300
|
|||
|
|
self._figlinewidth= 1.0
|
|||
|
|
|
|||
|
|
self._xfactor= 10.0 / 230.0 # x size * x factor = x length
|
|||
|
|
self._yfactor= 0.3 # y size * y factor = y length
|
|||
|
|
|
|||
|
|
jobstat= pdata[u'任务描述']
|
|||
|
|
|
|||
|
|
self._xsize_left= 12.0 # left blank
|
|||
|
|
self._xsize_right= 12.0 # right blank
|
|||
|
|
self._ysize_top= 0.3 # top blank
|
|||
|
|
self._ysize_bottom= 1.2 # bottom blank
|
|||
|
|
|
|||
|
|
self._ysize_gap1= 0.2
|
|||
|
|
self._ysize_gap2= 0.3 if (jobstat[u'历史价格子图'] or jobstat[u'历史换手率子图'] or jobstat[u'财务指标子图']) else 0.0
|
|||
|
|
|
|||
|
|
# 建立 X 轴参数
|
|||
|
|
#===============================================================================================================
|
|||
|
|
if jobstat[u'价格子图'] or jobstat[u'换手率子图']:
|
|||
|
|
xparams= {'xmargin': 1}
|
|||
|
|
xparams.update(self._compute_xparams()) # 与 X 轴坐标点相关的数据结构
|
|||
|
|
|
|||
|
|
if jobstat[u'历史价格子图'] or jobstat[u'历史换手率子图'] or jobstat[u'财务指标子图']:
|
|||
|
|
xparams_fs= {'xmargin': 3}
|
|||
|
|
xparams_fs.update(self._compute_xparams_fullspan())
|
|||
|
|
|
|||
|
|
# 建立子图对象
|
|||
|
|
#===============================================================================================================
|
|||
|
|
self._subplots= {}
|
|||
|
|
|
|||
|
|
if jobstat[u'公司信息子图']:
|
|||
|
|
name= 'basic'
|
|||
|
|
self._subplots[name]= SubPlot_BasicInfo(pdata=pdata, parent=self, name=name)
|
|||
|
|
|
|||
|
|
if jobstat[u'历史价格子图']: # XXX: 这个要放在 价格子图 前面，因为后者可能会用到它的 Y 轴坐标点位置
|
|||
|
|
name= 'pricefs'
|
|||
|
|
self._subplots[name]= SubPlot_PriceFullSpan(pdata=pdata, parent=self, xparams=xparams_fs, name=name)
|
|||
|
|
|
|||
|
|
if jobstat[u'价格子图']:
|
|||
|
|
name= 'price'
|
|||
|
|
self._subplots[name]= SubPlot_Price(pdata=pdata, parent=self, xparams=xparams, name=name)
|
|||
|
|
|
|||
|
|
if jobstat[u'财务指标子图']:
|
|||
|
|
name= 'financial'
|
|||
|
|
self._subplots[name]= SubPlot_Financial(pdata=pdata, parent=self, xparams=xparams_fs, name=name)
|
|||
|
|
|
|||
|
|
if jobstat[u'换手率子图']:
|
|||
|
|
name= 'torate'
|
|||
|
|
self._subplots[name]= SubPlot_TORate(pdata=pdata, parent=self, xparams=xparams, name=name)
|
|||
|
|
|
|||
|
|
if jobstat[u'历史换手率子图']:
|
|||
|
|
name= 'toratefs'
|
|||
|
|
self._subplots[name]= SubPlot_TORateFullSpan(pdata=pdata, parent=self, xparams=xparams_fs, name=name)
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
# 根据子图对象的尺寸计算自身的尺寸
|
|||
|
|
#===============================================================================================================
|
|||
|
|
self._xsize, \
|
|||
|
|
self._ysize= self._compute_size()
|
|||
|
|
|
|||
|
|
self._xlength= self._xsize * self._xfactor
|
|||
|
|
self._ylength= self._ysize * self._yfactor
|
|||
|
|
|
|||
|
|
# 根据计算出的尺寸建立 Figure 对象
|
|||
|
|
#===============================================================================================================
|
|||
|
|
self._Fig= pyplot.figure(figsize=(self._xlength, self._ylength), dpi=self._figdpi, facecolor=self._figfacecolor, \
|
|||
|
|
edgecolor=self._figedgecolor, linewidth=self._figlinewidth) # Figure 对象
|
|||
|
|
|
|||
|
|
# 用新建立的 Figure 对象交给子图对象，完成子图对象的初始化
|
|||
|
|
#===============================================================================================================
|
|||
|
|
rects= self._compute_rect()
|
|||
|
|
|
|||
|
|
|
|||
|
|
if 'basic' in self._subplots:
|
|||
|
|
self._subplots['basic'].build_axes(figobj=self._Fig, rect=rects['basic'])
|
|||
|
|
|
|||
|
|
# XXX: 这个要放在 price 前面，因为后者要用到它的 Axes 对象
|
|||
|
|
if 'torate' in self._subplots:
|
|||
|
|
self._subplots['torate'].build_axes(figobj=self._Fig, rect=rects['torate'])
|
|||
|
|
|
|||
|
|
if 'price' in self._subplots:
|
|||
|
|
self._subplots['price'].build_axes(figobj=self._Fig, rect=rects['price'])
|
|||
|
|
|
|||
|
|
# XXX: 这个要放在 pricefs 前面
|
|||
|
|
if 'toratefs' in self._subplots:
|
|||
|
|
self._subplots['toratefs'].build_axes(figobj=self._Fig, rect=rects['toratefs'])
|
|||
|
|
|
|||
|
|
if 'pricefs' in self._subplots:
|
|||
|
|
self._subplots['pricefs'].build_axes(figobj=self._Fig, rect=rects['pricefs'])
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
def _compute_size(self):
|
|||
|
|
'''
|
|||
|
|
根据子图的尺寸计算自身尺寸
|
|||
|
|
'''
|
|||
|
|
pdata= self._pdata
|
|||
|
|
jobstat= pdata[u'任务描述']
|
|||
|
|
|
|||
|
|
x_left, x_right= self._xsize_left, self._xsize_right
|
|||
|
|
y_top, y_bottom= self._ysize_top, self._ysize_bottom
|
|||
|
|
|
|||
|
|
y_gap1= self._ysize_gap1
|
|||
|
|
y_gap2= self._ysize_gap2
|
|||
|
|
|
|||
|
|
x_basic, y_basic= self._subplots['basic'].get_size() if 'basic' in self._subplots else (0.0, 0.0)
|
|||
|
|
x_price, y_price= self._subplots['price'].get_size() if 'price' in self._subplots else (0.0, 0.0)
|
|||
|
|
x_pricefs, y_pricefs= self._subplots['pricefs'].get_size() if 'pricefs' in self._subplots else (0.0, 0.0)
|
|||
|
|
x_torate, y_torate= self._subplots['torate'].get_size() if 'torate' in self._subplots else (0.0, 0.0)
|
|||
|
|
x_toratefs, y_toratefs= self._subplots['toratefs'].get_size() if 'toratefs' in self._subplots else (0.0, 0.0)
|
|||
|
|
x_financial, y_financial= self._subplots['financial'].get_size() if 'financial' in self._subplots else (0.0, 0.0)
|
|||
|
|
|
|||
|
|
x_all= x_left + max(x_price, x_basic, x_pricefs) + x_right
|
|||
|
|
y_all= y_top + y_basic + y_gap1 + y_pricefs + y_toratefs + y_financial + y_gap2 + y_price + y_torate + y_bottom
|
|||
|
|
|
|||
|
|
return (x_all, y_all)
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
def get_sizeset(self):
|
|||
|
|
sizeset= {
|
|||
|
|
'x': self._xsize,
|
|||
|
|
'y': self._ysize,
|
|||
|
|
'top': self._ysize_top,
|
|||
|
|
'bottom': self._ysize_bottom,
|
|||
|
|
'left': self._xsize_left,
|
|||
|
|
'right': self._xsize_right
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return sizeset
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
def _compute_rect(self):
|
|||
|
|
'''
|
|||
|
|
|
|||
|
|
'''
|
|||
|
|
pdata= self._pdata
|
|||
|
|
jobstat= pdata[u'任务描述']
|
|||
|
|
|
|||
|
|
x_left= self._xsize_left
|
|||
|
|
x_right= self._xsize_right
|
|||
|
|
y_top= self._ysize_top
|
|||
|
|
y_bottom= self._ysize_bottom
|
|||
|
|
x_all= self._xsize
|
|||
|
|
y_all= self._ysize
|
|||
|
|
|
|||
|
|
y_gap1= self._ysize_gap1 # basic 与 financial 之间的空隙
|
|||
|
|
y_gap2= self._ysize_gap2 # toratefs 与 price 之间的空隙
|
|||
|
|
|
|||
|
|
x_basic, y_basic= self._subplots['basic'].get_size() if 'basic' in self._subplots else (0.0, 0.0)
|
|||
|
|
x_price, y_price= self._subplots['price'].get_size() if 'price' in self._subplots else (0.0, 0.0)
|
|||
|
|
x_pricefs, y_pricefs= self._subplots['pricefs'].get_size() if 'pricefs' in self._subplots else (0.0, 0.0)
|
|||
|
|
x_torate, y_torate= self._subplots['torate'].get_size() if 'torate' in self._subplots else (0.0, 0.0)
|
|||
|
|
x_toratefs, y_toratefs= self._subplots['toratefs'].get_size() if 'toratefs' in self._subplots else (0.0, 0.0)
|
|||
|
|
x_financial, y_financial= self._subplots['financial'].get_size() if 'financial' in self._subplots else (0.0, 0.0)
|
|||
|
|
|
|||
|
|
rects= {}
|
|||
|
|
|
|||
|
|
if 'basic' in self._subplots:
|
|||
|
|
rect= ((x_left + (x_all-x_left-x_right-x_basic)/2) / x_all, (y_all - y_top - y_basic)/y_all, x_basic/x_all, y_basic/y_all) # K线图部分
|
|||
|
|
rects['basic']= rect
|
|||
|
|
|
|||
|
|
if 'price' in self._subplots:
|
|||
|
|
rect= ((x_left + (x_all-x_left-x_right-x_price)/2) / x_all, (y_bottom + y_torate)/y_all, x_price/x_all, y_price/y_all) # K线图部分
|
|||
|
|
rects['price']= rect
|
|||
|
|
|
|||
|
|
if 'torate' in self._subplots:
|
|||
|
|
rect= ((x_left + (x_all-x_left-x_right-x_torate)/2)/x_all, y_bottom/y_all, x_torate/x_all, y_torate/y_all) # 成交量部分
|
|||
|
|
rects['torate']= rect
|
|||
|
|
|
|||
|
|
if 'pricefs' in self._subplots:
|
|||
|
|
rect= ((x_left + (x_all-x_left-x_right-x_pricefs)/2)/x_all, (y_all - y_top - y_basic - y_gap1 - y_pricefs)/y_all, x_pricefs/x_all, y_pricefs/y_all)
|
|||
|
|
rects['pricefs']= rect
|
|||
|
|
|
|||
|
|
if 'toratefs' in self._subplots:
|
|||
|
|
rect= ((x_left + (x_all-x_left-x_right-x_toratefs)/2)/x_all, (y_bottom + y_torate + y_price + y_gap2)/y_all, x_toratefs/x_all, y_toratefs/y_all)
|
|||
|
|
rects['toratefs']= rect
|
|||
|
|
|
|||
|
|
return rects
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
def _compute_xparams(self):
|
|||
|
|
'''
|
|||
|
|
主要坐标点是每月第一个交易日，辅助坐标点是每周第一个交易日
|
|||
|
|
'''
|
|||
|
|
quotes= self._pdata[u'行情']
|
|||
|
|
sidx= self._pdata[u'任务描述'][u'起始偏移']
|
|||
|
|
eidx= self._pdata[u'任务描述'][u'结束偏移']
|
|||
|
|
|
|||
|
|
# 设定 X 轴上的坐标
|
|||
|
|
#==================================================================================================================================================
|
|||
|
|
datelist= [ datetime.date(int(ys), int(ms), int(ds)) for ys, ms, ds in [ dstr.split('-') for dstr in quotes[u'日期'][sidx:eidx] ] ]
|
|||
|
|
|
|||
|
|
# 确定 X 轴的 MajorLocator
|
|||
|
|
mdindex= [] # 每个月第一个交易日在所有日期列表中的 index
|
|||
|
|
allyears= set([d.year for d in datelist]) # 所有的交易年份
|
|||
|
|
|
|||
|
|
for yr in sorted(allyears):
|
|||
|
|
allmonths= set([d.month for d in datelist if d.year == yr]) # 当年所有的交易月份
|
|||
|
|
for mon in sorted(allmonths):
|
|||
|
|
monthday= min([dt for dt in datelist if dt.year==yr and dt.month==mon]) # 当月的第一个交易日
|
|||
|
|
mdindex.append(datelist.index(monthday))
|
|||
|
|
|
|||
|
|
xMajorLocator= FixedLocator(numpy.array(mdindex))
|
|||
|
|
|
|||
|
|
# 确定 X 轴的 MinorLocator
|
|||
|
|
wdindex= {} # value: 每周第一个交易日在所有日期列表中的 index; key: 当周的序号 week number（当周是第几周）
|
|||
|
|
|
|||
|
|
for d in datelist:
|
|||
|
|
isoyear, weekno= d.isocalendar()[0:2]
|
|||
|
|
dmark= isoyear*100 + weekno
|
|||
|
|
if dmark not in wdindex:
|
|||
|
|
wdindex[dmark]= datelist.index(d)
|
|||
|
|
|
|||
|
|
wdindex= sorted(wdindex.values())
|
|||
|
|
|
|||
|
|
xMinorLocator= FixedLocator(numpy.array(wdindex))
|
|||
|
|
|
|||
|
|
# 确定 X 轴的 MajorFormatter 和 MinorFormatter
|
|||
|
|
def x_major_formatter(idx, pos=None):
|
|||
|
|
return datelist[idx].strftime('%Y-%m-%d')
|
|||
|
|
|
|||
|
|
def x_minor_formatter(idx, pos=None):
|
|||
|
|
return datelist[idx].strftime('%m-%d')
|
|||
|
|
|
|||
|
|
xMajorFormatter= FuncFormatter(x_major_formatter)
|
|||
|
|
xMinorFormatter= FuncFormatter(x_minor_formatter)
|
|||
|
|
|
|||
|
|
return {'xMajorLocator': xMajorLocator,
|
|||
|
|
'xMinorLocator': xMinorLocator,
|
|||
|
|
'xMajorFormatter': xMajorFormatter,
|
|||
|
|
'xMinorFormatter': xMinorFormatter,
|
|||
|
|
'mdindex': mdindex,
|
|||
|
|
'wdindex': wdindex
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
def _compute_xparams_fullspan(self):
|
|||
|
|
'''
|
|||
|
|
主要坐标点是每季第一个交易日，辅助坐标点是每月第一个交易日。是给宏观子图用的。
|
|||
|
|
'''
|
|||
|
|
quotes= self._pdata[u'行情']
|
|||
|
|
|
|||
|
|
datelist= [ datetime.date(int(ys), int(ms), int(ds)) for ys, ms, ds in [ dstr.split('-') for dstr in quotes[u'日期'] ] ]
|
|||
|
|
|
|||
|
|
# 确定 X 轴的 MinorLocator
|
|||
|
|
mdindex= [] # 每个月第一个交易日在所有日期列表中的 index
|
|||
|
|
sdindex= [] # 每季度第一个交易日在所有日期列表中的 index
|
|||
|
|
ydindex= [] # 每年第一个交易日在所有日期列表中的 index
|
|||
|
|
|
|||
|
|
allyears= set([d.year for d in datelist]) # 所有的交易年份
|
|||
|
|
|
|||
|
|
for yr in sorted(allyears):
|
|||
|
|
allmonths= set([d.month for d in datelist if d.year == yr]) # 当年所有的交易月份
|
|||
|
|
for mon in sorted(allmonths):
|
|||
|
|
monthday= min([dt for dt in datelist if dt.year==yr and dt.month==mon]) # 当月的第一个交易日
|
|||
|
|
idx= datelist.index(monthday)
|
|||
|
|
|
|||
|
|
if mon in (1, 4, 7, 10):
|
|||
|
|
sdindex.append(idx)
|
|||
|
|
|
|||
|
|
if mon == 1:
|
|||
|
|
ydindex.append(idx)
|
|||
|
|
else:
|
|||
|
|
mdindex.append(idx)
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
xMajorLocator= FixedLocator(numpy.array(sdindex))
|
|||
|
|
xMinorLocator= FixedLocator(numpy.array(mdindex))
|
|||
|
|
|
|||
|
|
# 确定 X 轴的 MajorFormatter 和 MinorFormatter
|
|||
|
|
def x_major_formatter(idx, pos=None):
|
|||
|
|
return datelist[idx].strftime('%Y-%m-%d')
|
|||
|
|
|
|||
|
|
def x_minor_formatter(idx, pos=None):
|
|||
|
|
return datelist[idx].strftime('%m-%d')
|
|||
|
|
|
|||
|
|
xMajorFormatter= FuncFormatter(x_major_formatter)
|
|||
|
|
xMinorFormatter= FuncFormatter(x_minor_formatter)
|
|||
|
|
|
|||
|
|
return {'xMajorLocator': xMajorLocator,
|
|||
|
|
'xMinorLocator': xMinorLocator,
|
|||
|
|
'xMajorFormatter': xMajorFormatter,
|
|||
|
|
'xMinorFormatter': xMinorFormatter,
|
|||
|
|
'sdindex': sdindex,
|
|||
|
|
'mdindex': mdindex,
|
|||
|
|
'ydindex': ydindex
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
def plot(self):
|
|||
|
|
'''
|
|||
|
|
'''
|
|||
|
|
# self.plot_title()
|
|||
|
|
|
|||
|
|
# 调用子图对象的绘图函数
|
|||
|
|
if 'basic' in self._subplots:
|
|||
|
|
self._subplots['basic'].plot()
|
|||
|
|
|
|||
|
|
if 'price' in self._subplots:
|
|||
|
|
self._subplots['price'].plot()
|
|||
|
|
|
|||
|
|
if 'torate' in self._subplots:
|
|||
|
|
self._subplots['torate'].plot()
|
|||
|
|
|
|||
|
|
if 'pricefs' in self._subplots:
|
|||
|
|
self._subplots['pricefs'].plot()
|
|||
|
|
|
|||
|
|
if 'toratefs' in self._subplots:
|
|||
|
|
self._subplots['toratefs'].plot()
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
def plot_title(self):
|
|||
|
|
'''
|
|||
|
|
绘制整个 Figure 的标题
|
|||
|
|
'''
|
|||
|
|
info= self._pdata[u'公司信息']
|
|||
|
|
figobj= self._Fig
|
|||
|
|
|
|||
|
|
# 整个 figure 的标题
|
|||
|
|
subtitle= (info[u'代码'] + ' ' if u'代码' in info else '') + info[u'简称']
|
|||
|
|
subtitle_2= (info[u'代码二'] + ' ' if u'代码二' in info else '') + info[u'简称二']
|
|||
|
|
|
|||
|
|
figobj.suptitle(subtitle + ' / ' + subtitle_2, fontsize=12, fontproperties=__font_properties__)
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
def savefig(self, figpath):
|
|||
|
|
'''
|
|||
|
|
保存图片
|
|||
|
|
'''
|
|||
|
|
self._Fig.savefig(figpath, dpi=self._figdpi, facecolor=self._figfacecolor, edgecolor=self._figedgecolor, linewidth=self._figlinewidth)
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
if __name__ == '__main__':
|
|||
|
|
|
|||
|
|
# pfile 指明存放绘图数据的 pickle file，figpath 指定图片需存放的路径
|
|||
|
|
pfile= sys.argv[1]
|
|||
|
|
figpath= sys.argv[2]
|
|||
|
|
|
|||
|
|
# 绘图数据 pdata
|
|||
|
|
fileobj= open(name=pfile, mode='rb')
|
|||
|
|
pdata= pickle.load(fileobj)
|
|||
|
|
fileobj.close()
|
|||
|
|
os.remove(pfile)
|
|||
|
|
|
|||
|
|
myfig= MyFigure(pdata=pdata)
|
|||
|
|
myfig.plot()
|
|||
|
|
myfig.savefig(figpath=figpath)
|