99 lines
3.2 KiB
Python
99 lines
3.2 KiB
Python
|
from pyecharts.charts import Line
|
|||
|
|
from pyecharts import options as opts
|
|||
|
|
|
|||
|
|
from utils.comm import *
|
|||
|
|
|
|||
|
|
|
|||
|
|
class LimitList:
|
|||
|
|
|
|||
|
|
def __init__(self, trade_date):
|
|||
|
|
self.trade_date = trade_date.strftime('%Y%m%d')
|
|||
|
|
self.start_date = return_trading_day(trade_date, -7)
|
|||
|
|
self.end_date = self.trade_date
|
|||
|
|
self.limit_df = pd.DataFrame()
|
|||
|
|
for _ in range(5):
|
|||
|
|
try:
|
|||
|
|
# 涨跌停列表
|
|||
|
|
self.limit_df = xcsc_pro.limit_list_d(start_date=self.start_date, end_date=self.end_date)
|
|||
|
|
break
|
|||
|
|
except Exception as e:
|
|||
|
|
print("涨跌停列表数据获取失败，", e)
|
|||
|
|
time.sleep(1)
|
|||
|
|
continue
|
|||
|
|
else:
|
|||
|
|
print("涨跌停列表数据获取失败次数过多，请关注！！！")
|
|||
|
|
|
|||
|
|
def limit_list_chart(self):
|
|||
|
|
# 按照 trade_date 和 limit 进行分组统计
|
|||
|
|
grouped = self.limit_df.groupby(['trade_date', 'limit']).size().reset_index(name='count')
|
|||
|
|
|
|||
|
|
# 使用 pivot_table 将 limit 类型作为列，统计数量
|
|||
|
|
pivot_table = grouped.pivot_table(index='trade_date', columns='limit', values='count', fill_value=0)
|
|||
|
|
# print(pivot_table)
|
|||
|
|
# D：跌停，U：涨停，Z：炸板
|
|||
|
|
# 获取不同 limit 类型的数据
|
|||
|
|
u_list = pivot_table['U'].tolist()
|
|||
|
|
d_list = pivot_table['D'].tolist()
|
|||
|
|
z_list = pivot_table['Z'].tolist()
|
|||
|
|
trade_dates = pivot_table.index.tolist()
|
|||
|
|
|
|||
|
|
c = (
|
|||
|
|
Line(init_opts=opts.InitOpts(width="100%", height="380px"))
|
|||
|
|
.add_xaxis(trade_dates)
|
|||
|
|
.add_yaxis("跌停数", d_list)
|
|||
|
|
.add_yaxis("涨停数", u_list)
|
|||
|
|
.add_yaxis("炸板数", z_list)
|
|||
|
|
.set_global_opts(
|
|||
|
|
tooltip_opts=opts.TooltipOpts(trigger="axis"),
|
|||
|
|
title_opts=opts.TitleOpts(title="涨跌停 —— 复盘", pos_left="center"),
|
|||
|
|
yaxis_opts=opts.AxisOpts(offset=5),
|
|||
|
|
xaxis_opts=opts.AxisOpts(offset=5, axislabel_opts=opts.LabelOpts(rotate=45, interval=0)),
|
|||
|
|
legend_opts=opts.LegendOpts(pos_top="8%")
|
|||
|
|
)
|
|||
|
|
.render(f"../html/{self.trade_date}/limit_list_chart.html")
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
def html_page_data(self):
|
|||
|
|
"""
|
|||
|
|
生成HTML文件
|
|||
|
|
:return:
|
|||
|
|
"""
|
|||
|
|
print(self.limit_df)
|
|||
|
|
df = self.limit_df[self.limit_df['trade_date'] == self.end_date].fillna(0).sort_values(
|
|||
|
|
by=["industry", "limit_times"],
|
|||
|
|
ascending=False).reset_index(drop=True)
|
|||
|
|
|
|||
|
|
table_content = f'<table class="table table-hover"><thead>' \
|
|||
|
|
f'<tr>' \
|
|||
|
|
f'<th scope="col">股票代码</th>' \
|
|||
|
|
f'<th scope="col">所属行业</th>' \
|
|||
|
|
f'<th scope="col">股票名称</th>' \
|
|||
|
|
f'<th scope="col">换手率</th>' \
|
|||
|
|
f'<th scope="col">连板数</th>' \
|
|||
|
|
f'<th scope="col">状态</th>' \
|
|||
|
|
f'</tr>' \
|
|||
|
|
f'</thead>' \
|
|||
|
|
f'<tbody>'
|
|||
|
|
|
|||
|
|
for index, row in df.iterrows():
|
|||
|
|
table_content += f'<tr><th scope="row">{row["ts_code"]}</th>' \
|
|||
|
|
f'<td>{row["industry"]}</td>' \
|
|||
|
|
f'<td>{row["name"]}</td>' \
|
|||
|
|
f'<td>{row["turnover_ratio"]}</td>' \
|
|||
|
|
f'<td>{row["limit_times"]}</td>'
|
|||
|
|
if row['limit'] == 'D':
|
|||
|
|
table_content += f'<td class="table-success">跌停</td>'
|
|||
|
|
elif row['limit'] == 'U':
|
|||
|
|
table_content += f'<td class="table-danger">涨停</td>'
|
|||
|
|
elif row['limit'] == 'Z':
|
|||
|
|
table_content += f'<td class="table-warning">炸板</td>'
|
|||
|
|
table_content += '</tr>'
|
|||
|
|
table_content += f'</tbody></table>'
|
|||
|
|
return table_content
|
|||
|
|
|
|||
|
|
|
|||
|
|
if __name__ == '__main__':
|
|||
|
|
limit = LimitList(datetime.strptime('20231108', "%Y%m%d"))
|
|||
|
|
# limit.limit_list_chart()
|
|||
|
|
limit.html_page_data()
|