# 导入包
import numpy as np 
import pandas as pd 
import matplotlib.pyplot as plt 
import re 

from pyecharts.charts import Bar, Line, Pie, Page
from pyecharts import options as opts 
from pyecharts.globals import SymbolType, WarningType
WarningType.ShowWarning = False

import warnings
warnings.filterwarnings('ignore')

# 汇总数据
df_all = pd.read_excel('../data/辽宁交战汇总.xlsx')
df_all.head() 

# 技术数据
df_skill = pd.read_excel('../data/详细技术数据-修改版.xlsx')
df_skill.head()    

# 循环
df_skill2 = pd.DataFrame() 

for i in df_skill['场序'].drop_duplicates():
    # 前五行填1
    df_sel = df_skill[df_skill['场序'] == i]
    df_sel.iloc[:5, 0] = 1
    # 后面填0 
    df_sel.iloc[5:-1, 0] = 0
    # 追加
    df_skill2 = df_skill2.append(df_sel, ignore_index=True) 

df_skill2['首发'] = df_skill2['首发'].replace('总计', np.nan)
df_skill2.head()   

# 删除空值
df_time = df_skill2.dropna(axis=0)
df_time.head() 

# 计算时间
time_num = round(df_time.groupby('球员')['出场时间'].mean())
time_num = time_num.sort_values(ascending=False) 
time_num

球员
梅奥         40.0
郭艾伦        35.0
兰斯-史蒂芬森    35.0
赵继伟        33.0
韩德君        32.0
布兰登-巴斯     32.0
高诗岩        24.0
李晓旭        24.0
刘志轩        22.0
贺天举        17.0
丛明晨        15.0
王化东        11.0
郭旭          9.0
刘雁宇         7.0
鄢手骐         5.0
马壮          2.0
Name: 出场时间, dtype: float64

data_pair =  [list(z) for z in zip(time_num.index.tolist(), time_num.values.tolist())]

# 绘制饼图
pie1 = Pie(init_opts=opts.InitOpts(width='1350px', height='750px'))
pie1.add('', data_pair, radius=['35%', '60%'])
pie1.set_global_opts(title_opts=opts.TitleOpts(title='不同球员的平均出场时间'), 
                     legend_opts=opts.LegendOpts(is_show=False, orient='vertical', pos_top='15%', pos_left='2%'))
pie1.set_series_opts(label_opts=opts.LabelOpts(formatter="{b}:{c}min"))
pie1.render() 

# 合并数据
df_neiwai = pd.merge(df_skill2, df_all[['场序', '胜负']]) 

# 删除列
df_neiwai.dropna(inplace=True)

# 添加外援/国内球员标签
def tansform_label(x):
    if x == '梅奥' or x == '布兰登-巴斯' or x == '兰斯-史蒂芬森':
        return '外援'
    else:
        return '国内球员'

df_neiwai['标签1'] = df_neiwai.球员.apply(tansform_label)
df_neiwai.head() 

# 分组汇总
defen = df_neiwai.groupby(['场序', '标签1'])['得分'].mean().reset_index() 
defen['得分'] = round(defen['得分'])

# 产生数据
x_line1 = defen[defen['标签1']=='国内球员']['场序'].astype('str').tolist()
y_line1 = defen[defen['标签1']=='国内球员']['得分'].tolist()
y_line2 = defen[defen['标签1']=='外援']['得分'].tolist() 

line9 = Line(init_opts=opts.InitOpts(width='1350px', height='750px'))
line9.add_xaxis(x_line1) 
line9.add_yaxis('国内球员', y_line1,
                markpoint_opts=opts.MarkPointOpts(data=[
                    opts.MarkPointItem(type_='max', name='最大值'),
                    opts.MarkPointItem(type_='min', name='最小值')
                ]))
line9.add_yaxis('外援', y_line2,
                markpoint_opts=opts.MarkPointOpts(data=[
                    opts.MarkPointItem(type_='max', name='最大值'),
                    opts.MarkPointItem(type_='min', name='最小值')
                ]))
line9.set_global_opts(title_opts=opts.TitleOpts(title='外援-国内球员平均得分对比')) 
line9.set_series_opts(label_opts=opts.LabelOpts(is_show=True), 
                      linestyle_opts=opts.LineStyleOpts(width=3)  
                     ) 
line9.render()   

# 定义函数
def draw_line(xaxis_data, y_axis, title_label, series_name='', min_num=None, label_color=None, line_color=None, formatter=None):
    line = Line(init_opts=opts.InitOpts(width='1350px', height='750px'))
    line.add_xaxis(xaxis_data) 
    line.add_yaxis(series_name, y_axis,
                    markpoint_opts=opts.MarkPointOpts(data=[
                        opts.MarkPointItem(type_='max', name='最大值'),
                        opts.MarkPointItem(type_='min', name='最小值')
                    ])) 
    line.set_global_opts(title_opts=opts.TitleOpts(title=title_label), 
                          yaxis_opts=opts.AxisOpts(min_=min_num) 
                         ) 
    line.set_series_opts(label_opts=opts.LabelOpts(is_show=True, color=label_color, formatter=formatter), 
                          linestyle_opts=opts.LineStyleOpts(width=3, color=line_color)  
                         )
    return line

# 产生数据
x_line1 = df_all[df_all['胜负']=='胜']['场序'].astype('str').tolist()
y_line1 = df_all[df_all['胜负']=='胜']['得分'].tolist()

x_line2 = df_all[df_all['胜负']=='败']['场序'].astype('str').tolist()
y_line2 = df_all[df_all['胜负']=='败']['得分'].tolist() 

# 绘图
line1 = draw_line(xaxis_data=x_line1, y_axis=y_line1, title_label='全部比赛-胜场得分曲线图', min_num=80,
                 label_color=None, line_color=None)
line2 = draw_line(xaxis_data=x_line2, y_axis=y_line2, title_label='全部比赛-胜场负场曲线图', min_num=80, 
                  label_color='blue', line_color='#3B7BA9')

page1 = Page()  
page1.add(line1, line2)
page1.render()