项目周报汇报的时候要做数据汇总,总是要从不同的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()