# data = pd.concat([dates_year, dates_ratings, dates_votes], axis=1) df.columns = ['上映年份', '电影评分', '评分人数'] data = data.groupby('上映年份').agg({'上映年份': 'mean', '评分人数': 'sum'}).reset_index() # 绘制散点图 scatter = ( Scatter() .add_xaxis(data['上映年份'].tolist()) .add_yaxis("电影评分", data['电影评分'].tolist(), symbol_size=10, label_opts=opts.LabelOpts(is_show=False)) .add_yaxis("评分人数", data['评分人数'].tolist(), symbol_size=10, label_opts=opts.LabelOpts(is_show=False)) .set_global_opts( title_opts=opts.TitleOpts(title="电影年份评分人数散点图"), tooltip_opts=opts.TooltipOpts(trigger="axis", axis_pointer_type="cross"), xaxis_opts=opts.AxisOpts(name='上映年份', type_="category"), yaxis_opts=[ opts.AxisOpts(name='电影评分', type_="value"), opts.AxisOpts(name='评分人数', type_="value", position="right") ], datazoom_opts=[opts.DataZoomOpts(is_show=True, range_start=0, range_end=20, orient='horizontal')] ) ) scatter.render_notebook() 优化代码
时间: 2023-06-24 07:06:32 浏览: 126
重建WMSYS用户的WMSYS.WM_CONCAT函数的3个文件
# 使用链式调用
data = pd.concat([dates_year, dates_ratings, dates_votes], axis=1)
data.columns = ['上映年份', '电影评分', '评分人数']
data = data.groupby('上映年份').agg({'电影评分': 'mean', '评分人数': 'sum'}).reset_index()
# 绘制散点图
scatter = (
Scatter()
.add_xaxis(data['上映年份'].tolist())
.add_yaxis("电影评分", data['电影评分'].tolist(), symbol_size=10, label_opts=opts.LabelOpts(is_show=False))
.add_yaxis("评分人数", data['评分人数'].tolist(), symbol_size=10, label_opts=opts.LabelOpts(is_show=False))
.set_global_opts(
title_opts=opts.TitleOpts(title="电影年份评分人数散点图"),
tooltip_opts=opts.TooltipOpts(trigger="axis", axis_pointer_type="cross"),
xaxis_opts=opts.AxisOpts(name='上映年份', type_="category"),
yaxis_opts=[
opts.AxisOpts(name='电影评分', type_="value"),
opts.AxisOpts(name='评分人数', type_="value", position="right")
],
datazoom_opts=[opts.DataZoomOpts(is_show=True, range_start=0, range_end=20, orient='horizontal')]
)
)
scatter.render_notebook()
阅读全文