from pyecharts import options as opts # 全局、系列配置
from pyecharts.charts import Bar, Grid, Line # 柱形图
from pyecharts.globals import ThemeType # 主题风格
from pyecharts.charts import Kline, Line, Bar, Gauge, Pie, Scatter, TreeMap # K线图、线形图、柱状图、仪表盘、饼图、散点图、矩形树图
from pyecharts.charts import Grid, Tab # 组合组件、分页组件
from pyecharts.components import Table # 表格组件
import time # 时间模块
import pandas as pd # Pandas模块
import numpy as np # Numpy模块
from sqlalchemy import create_engine # 数据引擎模块
conn = create_engine('mysql+pymysql://root:test@127.0.0.1:3306/quant_db?charset=utf8') # 创建引擎
#################################################################################################################################
class Module:
'''
1、上证指数、涨跌幅
2、深证指数、涨跌幅
3、创业板指数、涨跌幅
'''
def __init__(self, sql):
self.sql = sql
self.data = pd.read_sql(self.sql, conn)
def func_kline(self, x, y, series, title,title_link,pos_left,**kwargs):
# 数据处理部分
x = self.data['trade_date'].tolist()
y = []
for index, row in self.data.iterrows():
list = [row['open'], row['close'], row['low'], row['high']]
y.append(list)
# K线图绘制
kline = (
Kline(init_opts=opts.InitOpts(theme=ThemeType.DARK))
.add_xaxis(x)
.add_yaxis(series, y)
.set_global_opts(
xaxis_opts=opts.AxisOpts(is_scale=True),
yaxis_opts=opts.AxisOpts(
is_scale=True,
splitarea_opts=opts.SplitAreaOpts(
is_show=True,
areastyle_opts=opts.AreaStyleOpts(opacity=1)
),
),
# 标题组件配置项
title_opts=opts.TitleOpts(
title=title, # 主标题文本
title_link=title_link, # 主标题跳转url链接,可以为None
pos_left='', # 标题组件离容器左侧的距离,可以为None,具体值，百分比，‘left、right、center’
pos_right='',
pos_top = '',
pos_bottom=''
),
# 图例组件配置项
legend_opts=opts.LegendOpts(
pos_left='', # 图例组件离容器左侧的距离,可以为None,具体值，百分比，‘left、right、center’
pos_right='',
pos_top='',
pos_bottom='',
orient='', # 图例列表的布局朝向。可选：'horizontal', 'vertical'
legend_icon='' # 图例项的 icon。标记类型包括 'circle', 'rect', 'roundRect', 'triangle', 'diamond', 'pin', 'arrow', 'none'
),
# 工具箱组件配置项
toolbox_opts=opts.ToolBoxFeatureSaveAsImagesOpts(
type_='', # 保存的图片格式。支持 'png' 和 'jpeg'。
is_show='', # 是否显示该工具。
title= "保存为图片", # 提示语
),
# 系列提示框配置项
tooltip_opts=opts.TooltipOpts(
)
)
)
)
def func_bar(self, x, y, series, title):
# 柱形图绘制
bar = (
Bar(init_opts=opts.InitOpts(theme=ThemeType.DARK))
.add_xaxis(x)
.add_yaxis(series, y)
.set_global_opts(
title_opts=opts.TitleOpts(title=title, pos_top="48%"),
legend_opts=opts.LegendOpts(pos_top="48%")
)
)
grid = (
Grid()
.add(kline, grid_opts=opts.GridOpts(pos_bottom="60%", pos_right='50%'))
.add(bar, grid_opts=opts.GridOpts(pos_top="60%"))
# .render("grid_vertical.html")
)
return grid
def func02(self):
# 1、近90日深证指数走势及涨跌幅
data = pd.read_sql(sql, conn)
x = data['trade_date'].tolist()
y1 = []
y2 = data['pct_chg'].tolist()
for index, row in data.iterrows():
list = [row['open'], row['close'], row['low'], row['high']]
print(list)
y1.append(list)
kline = (
Kline()
.add_xaxis(x)
.add_yaxis("深证指数", y1)
.set_global_opts(
xaxis_opts=opts.AxisOpts(is_scale=True),
yaxis_opts=opts.AxisOpts(
is_scale=True,
splitarea_opts=opts.SplitAreaOpts(
is_show=True, areastyle_opts=opts.AreaStyleOpts(opacity=1)
),
),
title_opts=opts.TitleOpts(title="近90日深证指数走势图"),
)
)
bar = (
Bar()
.add_xaxis(x)
.add_yaxis("深证涨跌幅", y2)
.set_global_opts(
title_opts=opts.TitleOpts(title="Bar-Brush示例", subtitle="我是副标题", pos_top="48%"),
legend_opts=opts.LegendOpts(pos_top="48%")
)
)
grid = (
Grid()
.add(kline, grid_opts=opts.GridOpts(pos_bottom="60%", pos_right='50%'))
.add(bar, grid_opts=opts.GridOpts(pos_top="60%"))
# .render("grid_vertical.html")
)
return grid
def func03(self):
# 1、近90日创业板指数走势及涨跌幅
data = pd.read_sql(sql, conn)
x = data['trade_date'].tolist()
y1 = []
y2 = data['pct_chg'].tolist()
for index, row in data.iterrows():
list = [row['open'], row['close'], row['low'], row['high']]
print(list)
y1.append(list)
kline = (
Kline()
.add_xaxis(x)
.add_yaxis("创业板指数", y1)
.set_global_opts(
xaxis_opts=opts.AxisOpts(is_scale=True),
yaxis_opts=opts.AxisOpts(
is_scale=True,
splitarea_opts=opts.SplitAreaOpts(
is_show=True, areastyle_opts=opts.AreaStyleOpts(opacity=1)
),
),
title_opts=opts.TitleOpts(title="近90日创业板指数走势图"),
)
)
bar = (
Bar()
.add_xaxis(x)
.add_yaxis("创业板涨跌幅", y2)
.set_global_opts(
title_opts=opts.TitleOpts(title="Bar-Brush示例", subtitle="我是副标题", pos_top="48%"),
legend_opts=opts.LegendOpts(pos_top="48%")
)
)
def func_grid(self):
grid = (
Grid()
.add(kline, grid_opts=opts.GridOpts(pos_bottom="60%", pos_right='50%'))
.add(bar, grid_opts=opts.GridOpts(pos_top="60%"))
# .render("grid_vertical.html")
)
return grid
def main():
# 上证指数
sql1 = '''
SELECT
trade_date, -- 交易日期
`open`, -- 开盘价
high, -- 最高价
low, -- 最低价
`close`, -- 收盘价
pct_chg -- 涨跌幅
FROM
stock_daily
WHERE
ts_code = '000001.SH'
'''
# 深证指数
sql2 = '''
SELECT
trade_date, -- 交易日期
`open`, -- 开盘价
high, -- 最高价
low, -- 最低价
`close`, -- 收盘价
pct_chg -- 涨跌幅
FROM
stock_daily
WHERE
ts_code = '399001.SH'
'''
# 创业板指数
sql3 = '''
SELECT
trade_date, -- 交易日期
`open`, -- 开盘价
high, -- 最高价
low, -- 最低价
`close`, -- 收盘价
pct_chg -- 涨跌幅
FROM
stock_daily
WHERE
ts_code = '399006.SH'
'''
if __name__ == '__main__':
grid01 = func01()
grid02
tab = Tab()
tab.add(grid01, "bar-example")
# tab.add(line_markpoint(), "line-example")
# tab.add(pie_rosetype(), "pie-example")
# tab.add(grid_mutil_yaxis(), "grid-example")
tab.render("tab_base.html")
print("完成")
