def strategy():
form = request.form
getStock = form.get('stock')
stock = getStock if getStock else "usTSLA"
getCircle = form.get('circle')
circle = getCircle if getCircle else "120"
buyStrategy = BuyTrendStrategy(stock, int(circle), 1000000)
closeDayList, closeValueList = buyStrategy.get_close_line()
buyDayList, buyValueList = buyStrategy.get_buy_point()
closeLine = Line("趋势购买策略", width="400", height="300")
closeLine.add(stock, closeDayList, closeValueList)
buyES = EffectScatter("购买点", width="400", height="300")
buyES.add("购买点",
buyDayList,
buyValueList,
symbol_size=8,
effect_scale=3,
effect_period=3,
symbol="triangle")
overlap = Overlap()
overlap.add(closeLine)
overlap.add(buyES)
return render_template(
"strategy.html",
strategyEchart=overlap.render_embed(),
host=REMOTE_HOST,
script_list=buyES.get_js_dependencies(),
)
def test_grid_four_direction():
v1 = [5, 20, 36, 10, 75, 90]
v2 = [10, 25, 8, 60, 20, 80]
bar = Bar("柱状图示例", height=720, width=1200, title_pos="65%")
bar.add("商家A", CLOTHES, v1, is_stack=True)
bar.add("商家B", CLOTHES, v2, is_stack=True, legend_pos="80%")
line = Line("折线图示例")
line.add("最高气温", WEEK, [11, 11, 15, 13, 12, 13, 10],
mark_point=["max", "min"], mark_line=["average"])
line.add("最低气温", WEEK, [1, -2, 2, 5, 3, 2, 0],
mark_point=["max", "min"], mark_line=["average"],
legend_pos="20%")
v1 = [5, 20, 36, 10, 75, 90]
v2 = [10, 25, 8, 60, 20, 80]
scatter = Scatter("散点图示例", title_top="50%", title_pos="65%")
scatter.add("scatter", v1, v2, legend_top="50%", legend_pos="80%")
es = EffectScatter("动态散点图示例", title_top="50%")
es.add("es", [11, 11, 15, 13, 12, 13, 10], [1, -2, 2, 5, 3, 2, 0],
effect_scale=6, legend_top="50%", legend_pos="20%")
grid = Grid()
grid.add(bar, grid_bottom="60%", grid_left="60%")
grid.add(line, grid_bottom="60%", grid_right="60%")
grid.add(scatter, grid_top="60%", grid_left="60%")
grid.add(es, grid_top="60%", grid_right="60%")
grid.render()
def view(numList):
line = Line('折线图', width=2000)
atter = ['数据异常', '不及格', '及格', '良好', '优秀', '满分']
v1 = [numList[0], numList[1], numList[2], numList[3], numList[4], numList[5]]
line.add('最高成绩', atter, v1,
mark_point=['max'], # 标点最大值
mark_line=['average']) # 虚线位置是平均分
line.add('最低成绩', atter, v1,
mark_point=['min'],
legend_pos='20%')
es = EffectScatter() # 调用闪烁点
es.add('', atter, v1, effect_scale=8)
# 调用合并函数，再一个图表上输出
overlop = Overlap()
overlop.add(line)
overlop.add(es)
pie = Pie('饼图', title_pos='80%')
pie.add('', atter, v1,
radius=[60, 30], # 控制内外半径的
center=[65, 50],
legend_pos='80%',
legend_orient='vertical')
# 显示两个示例图的调用函数，不然会覆盖
grid = Grid()
grid.add(overlop, grid_right='50%')
grid.add(pie, grid_left='60%')
grid.render('abc.html')
def test_line_es():
v1 = [5, 20, 36, 10, 10, 100]
line = Line("line-EffectScatter 示例")
line.add("", CLOTHES, v1, is_random=True)
es = EffectScatter()
es.add("", CLOTHES, v1, effect_scale=8)
overlap = Overlap()
overlap.add(line)
overlap.add(es)
overlap.render()
def test_grid_left_right():
v1 = [5, 20, 36, 10, 75, 90]
v2 = [10, 25, 8, 60, 20, 80]
scatter = Scatter(width=1200)
scatter.add("散点图示例", v1, v2, legend_pos="70%")
es = EffectScatter()
es.add("动态散点图示例", [11, 11, 15, 13, 12, 13, 10], [1, -2, 2, 5, 3, 2, 0],
effect_scale=6, legend_pos="20%")
grid = Grid()
grid.add(scatter, grid_left="60%")
grid.add(es, grid_right="60%")
grid.render()
def effectscatterByInd(title, datas, size=None):
# import random
# data = [random.randint(0, 100) for _ in range(80)]
range_color = ['#313695', '#4575b4', '#74add1', '#abd9e9', '#e0f3f8', '#ffffbf',
'#fee090', '#fdae61', '#f46d43', '#d73027', '#a50026']
if size is None:
scatter = EffectScatter(
title,
width=320,
height=180,
title_pos="center",
# background_color='#f0f0f0',
)
else:
scatter = EffectScatter(
title,
width=size['width'],
height=size['height'],
title_pos="center",
# background_color='#f0f0f0',
)
for _val in datas:
_effect_scale = 3.
_symbol_size = 3.
if "high" in _val:
_effect_scale += 1.5
_symbol_size += 8.
elif "norm" in _val:
_effect_scale += 1.
_symbol_size += 5.
else:
logging.log(logging.WARN, u">>> effectscatterByInd: %s" % datas[_val]['label'])
scatter.add("", datas[_val]['x'], datas[_val]['y'],
is_visualmap=False,
visual_range_color=range_color,
effect_scale=_effect_scale,
symbol_size=_symbol_size,
)
scatter.options['toolbox']['show'] = False
scatter.options['xAxis'][0]['show'] = False
scatter.options['yAxis'][0]['show'] = False
return scatter.render_embed()
def effectscatter(title, datas, size=None):
# import random
# data = [random.randint(0, 100) for _ in range(80)]
range_color = ['#313695', '#4575b4', '#74add1', '#abd9e9', '#e0f3f8', '#ffffbf',
'#fee090', '#fdae61', '#f46d43', '#d73027', '#a50026']
if size is None:
scatter = EffectScatter(
title,
width=320,
height=180,
title_pos="center",
background_color='#f0f0f0',
)
else:
scatter = EffectScatter(
title,
width=size['width'],
height=size['height'],
title_pos="center",
background_color='#f0f0f0',
)
_scale = 1.5
for _data in datas:
if "v" not in _data:
scatter.add("", _data['x'], _data['y'],
is_visualmap=False,
visual_range_color=range_color,
mark_line=['average'],
mark_point=['max', 'min'],
effect_scale=_scale,
symbol_size=5,
)
_scale += 1.
else:
scatter.add("", _data['x'], _data['y'],
is_visualmap=False,
# visual_range_color=range_color,
mark_line=['average'],
mark_point=['max', 'min'],
effect_scale=1.5 + float(_data['v'])/40.,
symbol_size=3+_data['v']/3,
)
scatter.options['toolbox']['show'] = False
return scatter.render_embed()
def test_overlap_es_scatter():
v1 = [10, 20, 30, 40, 50, 60]
v2 = [30, 30, 30, 30, 30, 30]
v3 = [50, 50, 50, 50, 50, 50]
v4 = [10, 10, 10, 10, 10, 10]
es = EffectScatter("Scatter-EffectScatter 示例")
es.add("es", v1, v2)
scatter = Scatter()
scatter.add("scatter", v1, v3)
es_1 = EffectScatter()
es_1.add("es_1", v1, v4, symbol='pin', effect_scale=5)
overlap = Overlap()
overlap.add(es)
overlap.add(scatter)
overlap.add(es_1)
overlap.render()
def duotu2_fun():
attr = ["衬衫", "羊毛衫", "雪纺衫", "裤子", "高跟鞋", "袜子"]
v1 = [5, 20, 36, 10, 75, 90]
v2 = [10, 25, 8, 60, 20, 80]
bar = Bar("柱状图示例", title_pos="65%")
bar.add("商家A", attr, v1, is_stack=True)
bar.add("商家B", attr, v2, is_stack=True, legend_pos="80%")
line = Line("折线图示例")
attr = ["周一", "周二", "周三", "周四", "周五", "周六", "周日"]
line.add(
"最高气温",
attr,
[11, 11, 15, 13, 12, 13, 10],
mark_point=["max", "min"],
mark_line=["average"],
)
line.add(
"最低气温",
attr,
[1, -2, 2, 5, 3, 2, 0],
mark_point=["max", "min"],
mark_line=["average"],
legend_pos="20%",
)
v1 = [5, 20, 36, 10, 75, 90]
v2 = [10, 25, 8, 60, 20, 80]
scatter = Scatter("散点图示例", title_top="50%", title_pos="65%")
scatter.add("scatter", v1, v2, legend_top="50%", legend_pos="80%")
es = EffectScatter("动态散点图示例", title_top="50%")
es.add(
"es",
[11, 11, 15, 13, 12, 13, 10],
[1, -2, 2, 5, 3, 2, 0],
effect_scale=6,
legend_top="50%",
legend_pos="20%",
)
grid = Grid(height=720, width=1200)
grid.add(bar, grid_bottom="60%", grid_left="60%")
grid.add(line, grid_bottom="60%", grid_right="60%")
grid.add(scatter, grid_top="60%", grid_left="60%")
grid.add(es, grid_top="60%", grid_right="60%")
return grid
def test_effectscatter_default():
v1 = [10, 20, 30, 40, 50, 60]
v2 = [25, 20, 15, 10, 60, 33]
es = EffectScatter("动态散点图示例")
es.add("effectScatter", v1, v2)
es.render()
def test_effectscatter_multiple_symbol_type():
es = EffectScatter("动态散点图各种图形示例")
es.add("", [10], [10], symbol_size=20, effect_scale=3.5,
effect_period=3, symbol="pin")
es.add("", [20], [20], symbol_size=12, effect_scale=4.5,
effect_period=4, symbol="rect")
es.add("", [30], [30], symbol_size=30, effect_scale=5.5,
effect_period=5, symbol="roundRect")
es.add("", [40], [40], symbol_size=10, effect_scale=6.5,
effect_brushtype='fill', symbol="diamond")
es.add("", [50], [50], symbol_size=16, effect_scale=5.5,
effect_period=3, symbol="arrow")
es.add("", [60], [60], symbol_size=6, effect_scale=2.5,
effect_period=3, symbol="triangle")
es.render()
def test_page_grid_timeline_overlap():
# Grid
v1 = [5, 20, 36, 10, 75, 90]
v2 = [10, 25, 8, 60, 20, 80]
bar = Bar("柱状图示例", height=720, width=1200, title_pos="65%")
bar.add("商家A", CLOTHES, v1, is_stack=True)
bar.add("商家B", CLOTHES, v2, is_stack=True, legend_pos="80%")
line = Line("折线图示例")
line.add(
"最高气温",
WEEK,
[11, 11, 15, 13, 12, 13, 10],
mark_point=["max", "min"],
mark_line=["average"],
)
line.add(
"最低气温",
WEEK,
[1, -2, 2, 5, 3, 2, 0],
mark_point=["max", "min"],
mark_line=["average"],
legend_pos="20%",
)
v1 = [5, 20, 36, 10, 75, 90]
v2 = [10, 25, 8, 60, 20, 80]
scatter = Scatter("散点图示例", title_top="50%", title_pos="65%")
scatter.add("scatter", v1, v2, legend_top="50%", legend_pos="80%")
es = EffectScatter("动态散点图示例", title_top="50%")
es.add(
"es",
[11, 11, 15, 13, 12, 13, 10],
[1, -2, 2, 5, 3, 2, 0],
effect_scale=6,
legend_top="50%",
legend_pos="20%",
)
grid = Grid()
grid.add(bar, grid_bottom="60%", grid_left="60%")
grid.add(line, grid_bottom="60%", grid_right="60%")
grid.add(scatter, grid_top="60%", grid_left="60%")
grid.add(es, grid_top="60%", grid_right="60%")
# Timeline
bar_1 = Bar("2012 年销量", "数据纯属虚构")
bar_1.add("春季", CLOTHES, [randint(10, 100) for _ in range(6)])
bar_1.add("夏季", CLOTHES, [randint(10, 100) for _ in range(6)])
bar_1.add("秋季", CLOTHES, [randint(10, 100) for _ in range(6)])
bar_1.add("冬季", CLOTHES, [randint(10, 100) for _ in range(6)])
bar_2 = Bar("2013 年销量", "数据纯属虚构")
bar_2.add("春季", CLOTHES, [randint(10, 100) for _ in range(6)])
bar_2.add("夏季", CLOTHES, [randint(10, 100) for _ in range(6)])
bar_2.add("秋季", CLOTHES, [randint(10, 100) for _ in range(6)])
bar_2.add("冬季", CLOTHES, [randint(10, 100) for _ in range(6)])
bar_3 = Bar("2014 年销量", "数据纯属虚构")
bar_3.add("春季", CLOTHES, [randint(10, 100) for _ in range(6)])
bar_3.add("夏季", CLOTHES, [randint(10, 100) for _ in range(6)])
bar_3.add("秋季", CLOTHES, [randint(10, 100) for _ in range(6)])
bar_3.add("冬季", CLOTHES, [randint(10, 100) for _ in range(6)])
bar_4 = Bar("2015 年销量", "数据纯属虚构")
bar_4.add("春季", CLOTHES, [randint(10, 100) for _ in range(6)])
bar_4.add("夏季", CLOTHES, [randint(10, 100) for _ in range(6)])
bar_4.add("秋季", CLOTHES, [randint(10, 100) for _ in range(6)])
bar_4.add("冬季", CLOTHES, [randint(10, 100) for _ in range(6)])
bar_5 = Bar("2016 年销量", "数据纯属虚构", height=720, width=1200)
bar_5.add("春季", CLOTHES, [randint(10, 100) for _ in range(6)])
bar_5.add("夏季", CLOTHES, [randint(10, 100) for _ in range(6)])
bar_5.add("秋季", CLOTHES, [randint(10, 100) for _ in range(6)])
bar_5.add(
"冬季",
CLOTHES,
[randint(10, 100) for _ in range(6)],
is_legend_show=True,
)
timeline = Timeline(is_auto_play=True, timeline_bottom=0)
timeline.add(bar_1, "2012 年")
timeline.add(bar_2, "2013 年")
timeline.add(bar_3, "2014 年")
timeline.add(bar_4, "2015 年")
timeline.add(bar_5, "2016 年")
# Overlap
attr = ["{}月".format(i) for i in range(1, 13)]
v1 = [2.0, 4.9, 7.0, 23.2, 25.6, 76.7, 135.6, 162.2, 32.6, 20.0, 6.4, 3.3]
v2 = [2.6, 5.9, 9.0, 26.4, 28.7, 70.7, 175.6, 182.2, 48.7, 18.8, 6.0, 2.3]
v3 = [2.0, 2.2, 3.3, 4.5, 6.3, 10.2, 20.3, 23.4, 23.0, 16.5, 12.0, 6.2]
bar = Bar(height=720, width=1200)
bar.add("蒸发量", attr, v1)
bar.add("降水量", attr, v2, yaxis_formatter=" ml", yaxis_max=250)
line = Line()
line.add("平均温度", attr, v3, yaxis_formatter=" °C")
overlap = Overlap()
overlap.add(bar)
overlap.add(line, yaxis_index=1, is_add_yaxis=True)
page = Page()
page.add(grid)
page.add(timeline)
page.add(overlap)
page.render()
def test_grid():
# grid_0
attr = ["衬衫", "羊毛衫", "雪纺衫", "裤子", "高跟鞋", "袜子"]
v1 = [5, 20, 36, 10, 75, 90]
v2 = [10, 25, 8, 60, 20, 80]
bar = Bar("柱状图示例", height=720, is_grid=True)
bar.add("商家A", attr, v1, is_stack=True, grid_bottom="60%")
bar.add("商家B", attr, v2, is_stack=True, grid_bottom="60%")
line = Line("折线图示例", title_top="50%")
attr = ['周一', '周二', '周三', '周四', '周五', '周六', '周日']
line.add("最高气温", attr, [11, 11, 15, 13, 12, 13, 10], mark_point=["max", "min"], mark_line=["average"])
line.add("最低气温", attr, [1, -2, 2, 5, 3, 2, 0], mark_point=["max", "min"],
mark_line=["average"], legend_top="50%")
bar.grid(line.get_series(), grid_top="60%")
bar.show_config()
bar.render()
# grid_1
v1 = [5, 20, 36, 10, 75, 90]
v2 = [10, 25, 8, 60, 20, 80]
scatter = Scatter(width=1200, is_grid=True)
scatter.add("散点图示例", v1, v2, grid_left="60%", legend_pos="70%")
es = EffectScatter()
es.add("动态散点图示例", [11, 11, 15, 13, 12, 13, 10], [1, -2, 2, 5, 3, 2, 0],
effect_scale=6, legend_pos="20%")
scatter.grid(es.get_series(), grid_right="60%")
scatter.show_config()
scatter.render()
# grid_2
attr = ["衬衫", "羊毛衫", "雪纺衫", "裤子", "高跟鞋", "袜子"]
v1 = [5, 20, 36, 10, 75, 90]
v2 = [10, 25, 8, 60, 20, 80]
bar = Bar("柱状图示例", height=720, width=1200, title_pos="65%", is_grid=True)
bar.add("商家A", attr, v1, is_stack=True, grid_bottom="60%", grid_left="60%")
bar.add("商家B", attr, v2, is_stack=True, grid_bottom="60%", grid_left="60%", legend_pos="80%")
line = Line("折线图示例")
attr = ['周一', '周二', '周三', '周四', '周五', '周六', '周日']
line.add("最高气温", attr, [11, 11, 15, 13, 12, 13, 10], mark_point=["max", "min"], mark_line=["average"])
line.add("最低气温", attr, [1, -2, 2, 5, 3, 2, 0], mark_point=["max", "min"],
mark_line=["average"], legend_pos="20%")
v1 = [5, 20, 36, 10, 75, 90]
v2 = [10, 25, 8, 60, 20, 80]
scatter = Scatter("散点图示例", title_top="50%", title_pos="65%")
scatter.add("scatter", v1, v2, legend_top="50%", legend_pos="80%")
es = EffectScatter("动态散点图示例", title_top="50%")
es.add("es", [11, 11, 15, 13, 12, 13, 10], [1, -2, 2, 5, 3, 2, 0], effect_scale=6,
legend_top="50%", legend_pos="20%")
bar.grid(line.get_series(), grid_bottom="60%", grid_right="60%")
bar.grid(scatter.get_series(), grid_top="60%", grid_left="60%")
bar.grid(es.get_series(), grid_top="60%", grid_right="60%")
bar.show_config()
bar.render()
# grid_3
line = Line("折线图示例", width=1200, is_grid=True)
attr = ['周一', '周二', '周三', '周四', '周五', '周六', '周日']
line.add("最高气温", attr, [11, 11, 15, 13, 12, 13, 10], mark_point=["max", "min"],
mark_line=["average"], grid_right="65%")
line.add("最低气温", attr, [1, -2, 2, 5, 3, 2, 0], mark_point=["max", "min"],
mark_line=["average"], legend_pos="20%")
attr = ["衬衫", "羊毛衫", "雪纺衫", "裤子", "高跟鞋", "袜子"]
v1 = [11, 12, 13, 10, 10, 10]
pie = Pie("饼图示例", title_pos="45%")
pie.add("", attr, v1, radius=[30, 55], legend_pos="65%", legend_orient='vertical')
line.grid(pie.get_series(), grid_left="60%")
line.show_config()
line.render()
# grid_4
line = Line("折线图示例", width=1200, is_grid=True)
attr = ['周一', '周二', '周三', '周四', '周五', '周六', '周日']
line.add("最高气温", attr, [11, 11, 15, 13, 12, 13, 10], mark_point=["max", "min"],
mark_line=["average"], grid_right="60%")
line.add("最低气温", attr, [1, -2, 2, 5, 3, 2, 0], mark_point=["max", "min"],
mark_line=["average"], legend_pos="20%", grid_right="60%")
attr = ["衬衫", "羊毛衫", "雪纺衫", "裤子", "高跟鞋", "袜子"]
value = [20, 40, 60, 80, 100, 120]
v1 = [[2320.26, 2320.26, 2287.3, 2362.94],
[2300, 2291.3, 2288.26, 2308.38],
[2295.35, 2346.5, 2295.35, 2345.92],
[2347.22, 2358.98, 2337.35, 2363.8],
[2360.75, 2382.48, 2347.89, 2383.76],
[2383.43, 2385.42, 2371.23, 2391.82],
[2377.41, 2419.02, 2369.57, 2421.15],
[2425.92, 2428.15, 2417.58, 2440.38],
[2411, 2433.13, 2403.3, 2437.42],
[2432.68, 2334.48, 2427.7, 2441.73],
[2430.69, 2418.53, 2394.22, 2433.89],
[2416.62, 2432.4, 2414.4, 2443.03],
[2441.91, 2421.56, 2418.43, 2444.8],
[2420.26, 2382.91, 2373.53, 2427.07],
[2383.49, 2397.18, 2370.61, 2397.94],
[2378.82, 2325.95, 2309.17, 2378.82],
[2322.94, 2314.16, 2308.76, 2330.88],
[2320.62, 2325.82, 2315.01, 2338.78],
[2313.74, 2293.34, 2289.89, 2340.71],
[2297.77, 2313.22, 2292.03, 2324.63],
[2322.32, 2365.59, 2308.92, 2366.16],
[2364.54, 2359.51, 2330.86, 2369.65],
[2332.08, 2273.4, 2259.25, 2333.54],
[2274.81, 2326.31, 2270.1, 2328.14],
[2333.61, 2347.18, 2321.6, 2351.44],
[2340.44, 2324.29, 2304.27, 2352.02],
[2326.42, 2318.61, 2314.59, 2333.67],
[2314.68, 2310.59, 2296.58, 2320.96],
[2309.16, 2286.6, 2264.83, 2333.29],
[2282.17, 2263.97, 2253.25, 2286.33],
[2255.77, 2270.28, 2253.31, 2276.22]]
kline = Kline("K 线图示例", title_pos="60%")
kline.add("日K", ["2017/7/{}".format(i + 1) for i in range(31)], v1, legend_pos="80%")
line.grid(kline.get_series(), grid_left="55%")
line.show_config()
line.render()
# grid_5
import random
x_axis = ["12a", "1a", "2a", "3a", "4a", "5a", "6a", "7a", "8a", "9a", "10a", "11a",
"12p", "1p", "2p", "3p", "4p", "5p", "6p", "7p", "8p", "9p", "10p", "11p"]
y_aixs = ["Saturday", "Friday", "Thursday", "Wednesday", "Tuesday", "Monday", "Sunday"]
data = [[i, j, random.randint(0, 50)] for i in range(24) for j in range(7)]
heatmap = HeatMap("热力图示例", height=700, is_grid=True)
heatmap.add("热力图直角坐标系", x_axis, y_aixs, data, is_visualmap=True, visual_top="45%",
visual_text_color="#000", visual_orient='horizontal', grid_bottom="60%")
attr = ["衬衫", "羊毛衫", "雪纺衫", "裤子", "高跟鞋", "袜子"]
v1 = [5, 20, 36, 10, 75, 90]
v2 = [10, 25, 8, 60, 20, 80]
bar = Bar("柱状图示例", title_top="52%")
bar.add("商家A", attr, v1, is_stack=True)
bar.add("商家B", attr, v2, is_stack=True, legend_top="50%")
heatmap.grid(bar.get_series(), grid_top="60%")
heatmap.show_config()
heatmap.render()
def test_effectscatter():
# effectSactter_0
v1 = [10, 20, 30, 40, 50, 60]
v2 = [25, 20, 15, 10, 60, 33]
es = EffectScatter("动态散点图示例")
es.add("effectScatter", v1, v2)
es.render()
# effectScatter_1
es = EffectScatter("动态散点图各种图形示例")
es.add("", [10], [10], symbol_size=20, effect_scale=3.5, effect_period=3, symbol="pin")
es.add("", [20], [20], symbol_size=12, effect_scale=4.5, effect_period=4, symbol="rect")
es.add("", [30], [30], symbol_size=30, effect_scale=5.5, effect_period=5, symbol="roundRect")
es.add("", [40], [40], symbol_size=10, effect_scale=6.5, effect_brushtype='fill', symbol="diamond")
es.add("", [50], [50], symbol_size=16, effect_scale=5.5, effect_period=3, symbol="arrow")
es.add("", [60], [60], symbol_size=6, effect_scale=2.5, effect_period=3, symbol="triangle")
es.render()
def sysplot(sys, new=True, axes=None, style=1):
"""绘制系统实际买入/卖出信号
:param SystemBase sys: 系统实例
:param new: 仅在未指定axes的情况下生效，当为True时，
创建新的窗口对象并在其中进行绘制
:param axes: 指定在那个轴对象中进行绘制
:param style: 1 | 2 信号箭头绘制样式
"""
kdata = sys.getTO()
refdates = kdata.getDatetimeList()
if kdata.getQuery().kType == KQuery.DAY:
x_list = [d.date() for d in refdates]
else:
x_list = [d.datetime() for d in refdates]
date_index = dict([(d,i) for i,d in enumerate(refdates)])
if axes is None:
if new:
axes = create_figure()
kplot(kdata, axes=axes)
else:
axes = gca()
es = EffectScatter()
highest = round(max(HIGH(kdata)),2)
lowest = round(min(LOW(kdata)), 2)
height = highest - lowest
tds = sys.tm.getTradeList()
buy_dates = []
sell_dates = []
for t in tds:
if t.business == BUSINESS.BUY:
buy_dates.append(t.datetime)
elif t.business == BUSINESS.SELL:
sell_dates.append(t.datetime)
else:
pass
dates = buy_dates
buy_y_list = ['-' for i in range(len(refdates))]
for d in dates:
if d not in date_index:
continue
pos = date_index[d]
krecord = kdata[pos]
buy_y_list[pos] = round(krecord.lowPrice - height*0.02, 2)
es.add("", x_list, buy_y_list,
symbol_size=12, effect_scale=2.5, effect_period=0,symbol="triangle",
is_label_show=True,
label_formatter='B',
label_pos = 'bottom',
label_text_color = '#CD0000',
label_color=['#CD0000', '#008B00'])
dates = sell_dates
sell_y_list = ['-' for i in range(len(refdates))]
for d in dates:
if d not in date_index:
continue
pos = date_index[d]
krecord = kdata[pos]
sell_y_list[pos] = round(krecord.highPrice + height*0.015, 2)
es.add("", x_list, sell_y_list,
symbol_size=20, effect_scale=2.5, effect_period=0,symbol="pin",
is_label_show=True,
label_formatter='S',
label_pos = 'top',
label_text_color = '#008B00',
label_color=['#CD0000', '#008B00', '#008B00'])
axes.add(es)
gcf().set_xaxis(x_list)
gcf().add_axis(axes)
return gcf()
def sgplot(sg, new=True, axes=None, style=1, kdata=None):
"""绘制买入/卖出信号
:param SignalBase sg: 信号指示器
:param new: 仅在未指定axes的情况下生效，当为True时，创建新的窗口对象并在其中进行绘制
:param axes: 指定在那个轴对象中进行绘制
:param style: 1 | 2 信号箭头绘制样式
:param KData kdata: 指定的KData（即信号发生器的交易对象），
如该值为None，则认为该信号发生器已经指定了交易对象，
否则，使用该参数作为交易对象
"""
if kdata is None:
kdata = sg.getTO()
else:
sg.setTO(kdata)
refdates = kdata.getDatetimeList()
if kdata.getQuery().kType == KQuery.DAY:
x_list = [d.date() for d in refdates]
else:
x_list = [d.datetime() for d in refdates]
date_index = dict([(d,i) for i,d in enumerate(refdates)])
if axes is None:
if new:
axes = create_figure()
kplot(kdata, axes=axes)
else:
axes = gca()
es = EffectScatter()
highest = round(max(HIGH(kdata)),2)
lowest = round(min(LOW(kdata)), 2)
height = highest - lowest
dates = sg.getBuySignal()
buy_y_list = ['-' for i in range(len(refdates))]
for d in dates:
if d not in date_index:
continue
pos = date_index[d]
krecord = kdata[pos]
buy_y_list[pos] = round(krecord.lowPrice - height*0.02, 2)
es.add("", x_list, buy_y_list,
symbol_size=12, effect_scale=2.5, effect_period=0,symbol="triangle",
is_label_show=True,
label_formatter='B',
label_pos = 'bottom',
label_text_color = '#CD0000',
label_color=['#CD0000', '#008B00'])
dates = sg.getSellSignal()
sell_y_list = ['-' for i in range(len(refdates))]
for d in dates:
if d not in date_index:
continue
pos = date_index[d]
krecord = kdata[pos]
sell_y_list[pos] = round(krecord.highPrice + height*0.015, 2)
es.add("", x_list, sell_y_list,
symbol_size=20, effect_scale=2.5, effect_period=0,symbol="pin",
is_label_show=True,
label_formatter='S',
label_pos = 'top',
label_text_color = '#008B00',
label_color=['#CD0000', '#008B00', '#008B00'])
axes.add(es)
gcf().set_xaxis(x_list)
gcf().add_axis(axes)
return gcf()
