首页 > > 网络编程 > 其它 >

Python: 读写Excel（openpyxl / win32com.client…

2019-04-11 10:10:15来源：博客园阅读 ()

项目周报汇报的时候要做数据汇总，总是要从不同的excel文件中去获取数据最后汇总到一个excel表里面，所以决定用python直接写个自动化脚本来自动执行。

用python来读写excel，目前找了2个方法：win32com.client和openpyxl

win32com.client可以对excel进行各种操作，可以调用VBA的库，相当于是模拟用户对excel进行操作，在执行过程中，你可以看到excel被打开，然后数据被写入，最后excel文件被关闭等等过程。（文档也可以参看OFFICE自带的VBA EXCEL 帮助文件(VBAXL.CHM)。这里面讲述了EXCEL VBA的编程概念，另外，《Python Programming on Win32》书中也有很详细的介绍。这本书中（第九章）给出了一个类来操作EXCEL 文件，可以很容易的加以扩展。）
openpyxl是处理excel2007/2010及以后的格式，也就是xlsx系列，如果要处理以前的2003的表格（xls），那么则要用另外的库(xlrd/xlwt等)。

python虽然并不是特别在意大小写，但是使用win32com.client一定要注意大小写，很多函数如果不区分大小写，是无法调用的，比如打开excel表格的Open函数，’O’必须大写，还有wb.Save()，‘S’也必须大写，而我们使用openpyxl使用小写即可。

openpyxl在保存时用save()，很多原有的格式图表是无法保留下来的，比如对excel进行修改，里边原有的透视表，用openpyxl的save()是无法保存的，但是使用win32com.client的wb.Save()却是可以保存这些图表的，这也是更加方便的地方。

这里简单分享项目中用到这2个模版的写数据方式：

首先已经把数据都存到字典中：

dict_data = {'ThinkPad_Users': 448177, 'IdeaPad_Users': 109626, 'Desktop_Users': 50605, 'Install_ThinkPad': 903036, 'Install_IdeaPad': 197467, 'Install_Desktop': 91656, 'ThinkPad_Fail': 8495, 'IdeaPad_Fail': 1970, 'Desktop_Fail': 1592}

然后需要做的是把字典中的数据写入到excel表格中：

使用win32com方法来修改Excel

import win32com.client

#使用win32com方法来修改Excel
def modify_excel_win32com(dict_data, filename, title):
    #用于修改Excel的配置
    xlApp = win32com.client.Dispatch('Excel.Application')

    #用xlApp打开用于修改和写入数据
    xlBook = xlApp.Workbooks.Open(filename, ReadOnly = False)

    sheet = xlBook.Worksheets('WeeklyData'
    col_size = sheet.UsedRange.columns.Count + 1
    #判断该title是否存在；如存在则覆盖数据；如不存在则新建数据
    print(sheet.UsedRange.Value[0])
    if title in sheet.UsedRange.Value[0]:
        # print(sheet.UsedRange.Value[0].index(title))
        col_size = sheet.UsedRange.Value[0].index(title) + 1
    else:
        try:
            col_size = sheet.UsedRange.Value[0].index(None) + 1
        except:
            pass
        finally:
            sheet.Cells(1, col_size).Value = title
            print(col_size)

    for key, value in dict_data.items():
        if key=='Install_A':
            sheet.Cells(2, col_size).Value = value
        elif key=='A_Users':
            sheet.Cells(3, col_size).Value = value
        elif key=='A_Fail':
            sheet.Cells(4, col_size).Value = value
            sheet.Cells(5, col_size).Value = '%.2f%%' % (value / sheet.Cells(2, col_size).Value * 100)
            # print(sheet.Cells(5, col_size).Value)
        elif key=='Install_B':
            sheet.Cells(6, col_size).Value = value
        elif key=='B_Users':
            sheet.Cells(7, col_size).Value = value
        elif key=='B_Fail':
            sheet.Cells(8, col_size).Value = value
            sheet.Cells(9, col_size).Value = '%.2f%%' % (value / sheet.Cells(6, col_size).Value * 100)
        elif key=='Install_C':
            sheet.Cells(10, col_size).Value = value
        elif key=='C_Users':
            sheet.Cells(11, col_size).Value = value
        elif key=='C_Fail':
            sheet.Cells(12, col_size).Value = value
            sheet.Cells(13, col_size).Value = '%.2f%%' % (value / sheet.Cells(10, col_size).Value * 100)
            sheet.Cells(14, col_size).Value = '%.2f%%' % ((sheet.Cells(4, col_size).Value + sheet.Cells(8, col_size).Value + value) 
                                                / (sheet.Cells(2, col_size).Value + sheet.Cells(6, col_size).Value + sheet.Cells(10, col_size).Value) 
                                                * 100)

    xlBook.Save()     #保存  
    xlApp.quit()   #关闭excel操作环境。

使用openpyxl库来修改Excel

from openpyxl import worksheet
from openpyxl import workbook
from openpyxl import load_workbook
from openpyxl.utils import get_column_letter

#使用openpyxl库来修改Excel
def modify_excel_openpyxl(dict_data, filename, title):    
    wb = load_workbook(filename)  #打开一个工作簿    
    sheet = wb['WeeklyData']  #获取一张表
    
    col_size = sheet.max_column+1
    first_row = []
    for i in range(1, col_size):
        first_row.append(sheet.cell(row=1, column=i).value)
    print(first_row)
    #判断该title是否存在；如存在则覆盖数据；如不存在则新建数据
    if title in first_row:
        col_size = first_row.index(title) + 1
    else:
        try:
            col_size = first_row.index(None) + 1
        except:
            pass
        finally:
            sheet.cell(row=1, column=col_size, value=title)

    print(get_column_letter(col_size))
    col_letter = get_column_letter(col_size)
    sheet[col_letter+'2'] = 'testtest'

    for key, value in dict_data.items():
        if key=='Install_A':
            sheet[col_letter+'2'] = value
        elif key=='A_Users':
            sheet[col_letter+'3'] = value
        elif key=='A_Fail':
            sheet[col_letter+'4'] = value
            sheet[col_letter+'5'] = '='+col_letter+'4/'+col_letter+'2'
        elif key=='Install_B':
            sheet[col_letter+'6'] = value
        elif key=='B_Users':
            sheet[col_letter+'7'] = value
        elif key=='B_Fail':
            sheet[col_letter+'8'] = value
            sheet[col_letter+'9'] = '='+col_letter+'8/'+col_letter+'6'
        elif key=='Install_C':
            sheet[col_letter+'10'] = value
        elif key=='C_Users':
            sheet[col_letter+'11'] = value
        elif key=='C_Fail':
            sheet[col_letter+'12'] = value
            sheet[col_letter+'13'] = '='+col_letter+'12/'+col_letter+'10'
            sheet[col_letter+'14'] = '=('+col_letter+'4+'+col_letter+'8+'+col_letter+'12)/('+col_letter+'2+'+col_letter+'6+'+col_letter+'10)'

    wb.save(filename)
    wb.close()