# -*- coding: utf-8 -*-

import xlsxwriter

workbook = xlsxwriter.Workbook('chart.xlsx')
worksheet = workbook.add_worksheet()

# 指定類型為柱狀圖
chart = workbook.add_chart({'type': 'column'})

title = [u'業(yè)務名稱',u'星期一',u'星期二',u'星期三',u'星期四',u'星期五',u'星期六',u'星期日',u'平均流量']
buname= [u'官網(wǎng)',u'直播',u'購物',u'視頻',u'下載']

data = [
  [150,152,158,149,155,145,148],
  [189,188,195,193,198,1100,199],
  [201,200,198,175,170,198,195],
  [75,77,78,78,74,70,79],
  [188,85,87,190,93,188,184],
]

# 定義format格式對象
format=workbook.add_format()
# 定義format對象單元格邊框加粗1像素
format.set_border(1)

format_title=workbook.add_format()
format_title.set_border(1)
# format_title對象單元格背景色為#cccccc
format_title.set_bg_color('#cccccc')
# 居中格式
format_title.set_align('center')
format_title.set_bold()

format_ave=workbook.add_format()
format_ave.set_border(1)
# 以小數(shù)形式顯示
format_ave.set_num_format('0.00')

# 以行或列的方式寫入數(shù)據(jù)，同時引用格式
worksheet.write_row('A1',title,format_title)
worksheet.write_column('A2', buname,format)
worksheet.write_row('B2', data[0],format)
worksheet.write_row('B3', data[1],format)
worksheet.write_row('B4', data[2],format)
worksheet.write_row('B5', data[3],format)
worksheet.write_row('B6', data[4],format)

# 定義圖表數(shù)據(jù)系列函數(shù)
def chart_series(cur_row):
  worksheet.write_formula('I'+cur_row, \
   '=AVERAGE(B'+cur_row+':H'+cur_row+')',format_ave) # 計算平均值
  chart.add_series({
    'categories': '=Sheet1!$B$1:$H$1', # 將周一到周日作為圖標x軸
    'values':   '=Sheet1!$B$'+cur_row+':$H$'+cur_row, # 一周所有數(shù)據(jù)作為數(shù)據(jù)區(qū)域
    'line':    {'color': 'black'}, # 線條顏色定義為黑色
    'name': '=Sheet1!$A$'+cur_row, # 引用名稱為圖例
  })

for row in range(2, 7): # 數(shù)據(jù)以2-6行進行圖表數(shù)據(jù)系列函數(shù)調(diào)用
  chart_series(str(row))

chart.set_table() # 設置x軸格式
chart.set_style(30) # 設置圖表樣式

# 設置圖表大小
chart.set_size({'width': 577, 'height': 287})
# 設置圖表上方標題
chart.set_title ({'name': u'業(yè)務流量周報圖表'})
# 設置y軸標題
chart.set_y_axis({'name': 'Mb/s'})

# 在A8插入單元格圖表
worksheet.insert_chart('A8', chart)
workbook.close()