112 lines
3.9 KiB
Python
112 lines
3.9 KiB
Python
from pyecharts.charts import Bar, Pie, Page
|
||
from pyecharts import options as opts
|
||
from pyecharts.commons.utils import JsCode
|
||
import os
|
||
|
||
def generate_chart(
|
||
_data,
|
||
title: str, # 图表标题
|
||
category_columns: list, # 分类数据列（X 轴或饼图标签）
|
||
value_column: str, # 数值数据列（Y 轴或饼图值）
|
||
output_file: str = "chart.html" # 输出文件路径
|
||
):
|
||
"""
|
||
统一生成柱形图或饼形图。
|
||
|
||
参数:
|
||
_data: 结果集，通常是一个列表（字典格式）。
|
||
chart_type (str): 图表类型，'bar' 或 'pie'。
|
||
title (str): 图表标题。
|
||
category_columns (list): 分类数据列（X 轴或饼图标签）。
|
||
value_column (str): 数值数据列（Y 轴或饼图值）。
|
||
output_file (str): 输出 HTML 文件名（默认 "chart.html"）。
|
||
"""
|
||
try:
|
||
# 如果结果集为空，直接返回
|
||
if not _data:
|
||
raise ValueError("结果集为空，无法生成图表！")
|
||
|
||
# 检查列名是否存在
|
||
for col in category_columns + [value_column]:
|
||
if col not in _data[0]:
|
||
raise ValueError(f"列名 '{col}' 不存在！")
|
||
|
||
# 提取分类数据（动态组合多列）
|
||
categories = [" - ".join(str(row[col]) for col in category_columns) for row in _data]
|
||
# 提取数值数据
|
||
values = [row[value_column] for row in _data]
|
||
|
||
# 颜色配置
|
||
colors = [
|
||
"#5470C6", "#91CC75", "#EE6666", "#73C0DE", "#3BA272", "#FC8452", "#9A60B4", "#EA7CCC"
|
||
]
|
||
color_js = JsCode(f'''
|
||
function(params) {{
|
||
var colorList = {colors};
|
||
return colorList[params.dataIndex % colorList.length];
|
||
}}
|
||
''')
|
||
|
||
# 生成柱形图
|
||
bar_chart = Bar()
|
||
bar_chart.add_xaxis(categories)
|
||
bar_chart.add_yaxis(
|
||
series_name=value_column,
|
||
y_axis=values,
|
||
bar_width="30%",
|
||
itemstyle_opts=opts.ItemStyleOpts(color=color_js)
|
||
)
|
||
bar_chart.set_global_opts(
|
||
title_opts=opts.TitleOpts(title=title),
|
||
xaxis_opts=opts.AxisOpts(
|
||
name=" - ".join(category_columns),
|
||
axislabel_opts=opts.LabelOpts(rotate=45),
|
||
boundary_gap=True,
|
||
splitline_opts=opts.SplitLineOpts(is_show=False)
|
||
),
|
||
yaxis_opts=opts.AxisOpts(name=value_column),
|
||
legend_opts=opts.LegendOpts(is_show=False)
|
||
)
|
||
|
||
# 生成饼形图
|
||
data_pairs = list(zip(categories, values))
|
||
pie_chart = Pie()
|
||
pie_chart.add(
|
||
series_name="占比",
|
||
data_pair=data_pairs,
|
||
radius=["30%", "55%"],
|
||
label_opts=opts.LabelOpts(
|
||
formatter="{b}\n{d}%",
|
||
position="outside",
|
||
font_size=14
|
||
),
|
||
itemstyle_opts=opts.ItemStyleOpts(color=color_js)
|
||
)
|
||
pie_chart.set_global_opts(
|
||
title_opts=opts.TitleOpts(
|
||
title=title,
|
||
subtitle="数据维度：" + " → ".join(category_columns)
|
||
),
|
||
legend_opts=opts.LegendOpts(is_show=False)
|
||
)
|
||
|
||
|
||
# 确保目标目录存在
|
||
os.makedirs(os.path.dirname(output_file), exist_ok=True)
|
||
|
||
# 创建 Page 实例，并指定标题
|
||
page = Page(page_title="长春云校数据统计分析系统")
|
||
# 将两个图表添加到 Page 实例中
|
||
page.add(bar_chart)
|
||
page.add(pie_chart)
|
||
# 将 Page 实例保存为 HTML 文件
|
||
page.render(output_file)
|
||
|
||
# 检查文件是否存在
|
||
if os.path.exists(output_file):
|
||
print("文件生成成功！")
|
||
else:
|
||
print("文件生成失败！")
|
||
|
||
except Exception as e:
|
||
print(f"生成图表时发生错误: {e}") |