from pyecharts.charts import EffectScatter from pyecharts.components import Table from pyecharts import options as opts from pyecharts.globals import CurrentConfig, NotebookType CurrentConfig.NOTEBOOK_TYPE = NotebookType.JUPYTER_LAB import pandas as pd from pyecharts.globals import ThemeType if __name__ == '__main__': user_info = pd.read_csv('user_info.txt', delimiter='\t') # 统计用户年龄和性别分布 age_sex_count = user_info.groupby(['age', 'sex']).size().reset_index(name='count') # 将数据处理成可用于绘制小提琴图的格式 data = [] sexes = ['M', 'F'] for sex in sexes: age_count = [ {'name': str(age), 'value': count} for age, count in age_sex_count.loc[age_sex_count['sex'] == sex, ['age', 'count']].values ] data.append(age_count) # 使用 EffectScatter 绘制小提琴图 violin_chart = ( EffectScatter(init_opts=opts.InitOpts(theme=ThemeType.LIGHT)) .add_xaxis(['男性', '女性']) .add_yaxis("", data) .set_global_opts( title_opts=opts.TitleOpts(title='用户年龄和性别分布'), visualmap_opts=opts.VisualMapOpts(is_piecewise=True, pieces=[ {'min': 0, 'max': 50, 'label': '0~50', 'color': '#7f1818'}, {'min': 50, 'max': 100, 'label': '50~100', 'color': '#e7ba52'}, {'min': 100, 'max': 150, 'label': '100~150', 'color': '#6a9f2a'}, {'min': 150, 'max': 200, 'label': '150~200', 'color': '#0065c4'}, ]), toolbox_opts=opts.ToolboxOpts(is_show=True, orient='vertical', pos_left='right', feature={ 'saveAsImage': {'title': '保存'} }) ) ) # 添加表格 table_data = age_sex_count.sort_values(by=['age', 'sex']).reset_index(drop=True) table = ( Table() .add(headers=table_data.columns.tolist(), rows=table_data.values.tolist()) .set_global_opts( title_opts=opts.ComponentTitleOpts(title='用户年龄和性别分布表格', subtitle=''), toolbox_opts=opts.ToolboxOpts( is_show=True, orient='vertical', pos_left='right', feature={ 'saveAsImage': {'title': '保存'} } ) ) ) # 将小提琴图和表格组成一个页面 page = ( violin_chart .overlap(table) .render('d.html') )