Python: 读写Excel(openpyxl / win32com.client)

项目周报汇报的时候要作数据汇总,老是要从不一样的excel文件中去获取数据最后汇总到一个excel表里面,因此决定用python直接写个自动化脚原本自动执行。html

用python来读写excel,目前找了2个方法:win32com.client和openpyxlpython

  • 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()倒是能够保存这些图表的,这也是更加方便的地方。app

这里简单分享项目中用到这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}ui

  • 而后须要作的是把字典中的数据写入到excel表格中:

使用win32com方法来修改Excelspa

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库来修改Excelexcel

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()
相关文章
相关标签/搜索