操作Excel的Python模块有很多,从操作难易度和功能两方面考虑,选择了下面两个模块。
1. 读Excel模块 xlrd
API reference
xlrd提供了按行读取和按单元格读取两种方式
下面是按行读取的一个sample
样例数据如下:
import xlrd
def read_input():
workbook = xlrd.open_workbook('王者荣耀.xlsx') #(1)取得excel book对象
s12 = workbook.sheet_by_name("S12战绩") #(2)取得sheet对象
rows = s12.nrows #(3)获得总行数
for r in range(1,rows):
row = s12.row_values(r) #(4)获取行数据
hero_nm, games, wins = row
print("召唤%s 苦战%d场 胜率为%.2f%%" % (hero_nm,games,float(wins)*100/games))
if __name__ == '__main__':
read_input()
Output:
PS C:\wsp\wzry> & python c:/wsp/wzry/wzry.py
召唤扁鹊 苦战19场 胜率42.11%
召唤典韦 苦战33场 胜率60.61%
召唤后裔 苦战40场 胜率82.50%
2. 写Excel模块 xlsxwriter
API reference
与读模块xlrd对应的还有一个写模块xlwt,不过和xlsxwriter相比,功能要弱一些,使用起来也没有xlsxwriter方便,下面是一个生成excel图表的简单例子。
import xlsxwriter
def write():
sample_date = [
['项羽', 400,95],
['后裔', 387, 206],
['刘备', 23, 80],
['典韦', 0, 206],
['扁鹊', 27, 169]
]
STYLE_HEADER = {'font_size':9, 'border':1, 'bold': 1, 'bg_color':'#B4C6E7', 'align':'center', 'valign':'vcenter' }
STYLE_TEXT = {'font_size':9, 'border':1}
STYLE_NUMBER = {'font_size':9, 'border':1, 'num_format':'0.00'}
workbook = xlsxwriter.Workbook('王者荣耀_report.xlsx')
style_header = workbook.add_format(STYLE_HEADER)
style_text = workbook.add_format(STYLE_TEXT)
style_number = workbook.add_format(STYLE_NUMBER)
team_sheet = workbook.add_worksheet("两赛季比较")
header = ["英雄","S11出场次数","S12出场次数"]
team_sheet.write_row('A1', header, style_header)
# 宽度设定
widths = [8,15,15]
for ind,wid in enumerate(widths):
team_sheet.set_column(ind,ind,wid)
for ind, data in enumerate(sample_date) :
team_sheet.write(ind + 1,0,data[0],style_text)
team_sheet.write(ind + 1,1,data[1],style_number)
team_sheet.write(ind + 1,2,data[2],style_number)
#取得最大行号
ind += 2
#生成柱状图
chart_col = workbook.add_chart({'type': 'column'})
chart_col.add_series({
'name': 'S11英雄出场次数',
'categories': '=两赛季比较!$A$2:$A$%d' % ind,
'values': '=两赛季比较!$B$2:$B$%d' % ind,
})
chart_col.add_series({
'name': 'S12英雄出场次数',
'categories': '=两赛季比较!$A$2:$A$%d' % ind,
'values': '=两赛季比较!$C$2:$C$%d' % ind,
})
#设置图表属性
chart_col.set_size({'width': (len(sample_date) * 90 + 100)})
chart_col.set_title({'name': '两赛季英雄出场次数比较'})
chart_col.set_x_axis({'name': '英雄'})
chart_col.set_y_axis({'name': '出场次数'})
team_sheet.insert_chart('E1', chart_col)
workbook.close()
if __name__ == '__main__':
write()
这里只是列举了两个简单的例子,没有将用法一一列举,使用时可查阅API reference