Python 操作Excel

2019年5月19日 114次阅读 来源: teitiyuu

操作Excel的Python模块有很多，从操作难易度和功能两方面考虑，选择了下面两个模块。

1. 读Excel模块 xlrd

API reference
xlrd提供了按行读取和按单元格读取两种方式
下面是按行读取的一个sample

样例数据如下：

《Python 操作Excel》 王者荣耀.xlsx中的sheet[S12战绩]

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()

《Python 操作Excel》 王者荣耀_report.xlsx

这里只是列举了两个简单的例子，没有将用法一一列举，使用时可查阅API reference

    原文作者：teitiyuu
    原文地址: https://www.jianshu.com/p/19878005e356
    本文转自网络文章，转载此文章仅为分享知识，如有侵权，请联系博主进行删除。
点赞