实际工作中,我们经常会把表格某个区域(如:A1:F5)或某个图形保存为图片,如何用python自动做到这一点?不知屏幕前的小伙伴有没有遇到过类似的需求,此刻脑海里有木有一丢丢思路。
python操作excel的第三方库有很多,个个都有各自的绝招和擅长的应用场景,简单罗列一下:
- pyexcel:pyexcel是一个用于读写Excel、CSV和其他文件格式的库,提供了简单易用的接口。
- pandas-ExcelWriter:这是Pandas库的一个模块,用于将数据写入Excel文件。
- DataNitro:DataNitro是一个插件,可以在Excel中使用Python脚本,实现数据处理和分析。
- pyxlsb:pyxlsb是一个用于读取Excel二进制文件(xlsb格式)的库。
- Pandas:Pandas是一个强大的数据处理库,可以读取、写入和操作Excel文件。
- OpenPyXL:OpenPyXL是一个专门用于读写Excel 2010 xlsx/xlsm/xltx/xltm文件的库。
- XlsxWriter:XlsxWriter是一个用于写入Excel文件的库,支持Excel 2007 xlsx文件格式。
- xlrd / xlwt:这两个库分别用于读取和写入Excel文件,支持Excel 2003以前的xls文件格式。
- pywin32(win32com.client):这个库可以显式(或隐式)地打开excel进行一列操作,语法基本完全沿用VBA中的语法,只需要稍加pythonic改造。
这些库提供了诸多的功能和特性,几乎涉及到表格处理的方方面面,可以根据项目需求选择最适合的库来处理Excel表格。具体到我们今天的场景,使用win32com.client是最优解,因为它可以用到vba中已经提供的图片类API,比如-复制-选择性粘贴-粘贴为图片。
需要说明的是,为了区域转图片成功,必须做到以下几点:
1、当前工作表处于选中状态;
2、当前区域必须至少部分肉眼可见;
3、当前表格的显示比例需要默认为100%。
为了避免截图后的图片是漆黑(无色)的底色,建议提前将对应单元格区域的背景色设置为白色或者其他颜色。
有了这些前提后,就可以用单元格区域的copyPicture()方法将单元格区域数据写入剪贴板,再借助PIL库的ImageGrab.grabclipboard()方法从剪贴板获取数据,并保存为图片啦。
先看看CopyPicture的语法,如何设置参数。
完整的python示例代码如下:
1 from PIL import ImageGrab
2 import win32com.client as win32
3 from win32api import RGB
4
5 def capture_sheet_range_to_picture(sheetFilePath,sheetName,sheetRange,imgFilePath):
6 '''从某个excel的某个表中的某个区域截图,并保存为图片
7 sheetRange:str,如:"A1:E100"
8 '''
9 # 启动Excel应用
10 xlApp = win32.Dispatch('Excel.Application')
11 xlApp.Visible=True
12 # 打开目标Excel文件
13 workbook = xlApp.Workbooks.Open(sheetFilePath)
14 # 选择指定的工作表
15 detailSheet = workbook.Sheets(sheetName)
16 xlApp.ActiveWindow.ScrollRow = 2 #将水平滚动条拉到最上方
17 xlApp.ActiveWindow.ScrollColumn = 2
18 xlApp.ActiveWindow.Zoom = 100 # 调整表格的缩放为100%显示
19
20 detailSheet.Range(sheetRange).Interior.Color = RGB(255,255,255) # 纯白色
21 time.sleep(0.3)
22 # 将区域复制为图片
23
24 try:
25 detailSheet.Range(sheetRange).CopyPicture(1,2)
26 img = ImageGrab.grabclipboard() # 获取图片数据
28 img.save(imgFilePath)
29 except Exception as e:
30 # 如果截图失败,则重试一次
31 time.sleep(0.3)
32 detailSheet.Range(sheetRange).CopyPicture(1,2)
33 img = ImageGrab.grabclipboard() # 获取图片数据
34 img.save(imgFilePath)
35 # 关闭工作簿并退出Excel
36 workbook.Close(SaveChanges=False)
37 xlApp.Quit()
38 return
这段代码【信息量】很大,您可别不信,都是小爬我一遍遍踩坑才得来的宝贵经验哟,相信您已经受到了不少启发,那就动手试试吧,一定会收获满满。
快来关注本公众号 获取更多爬虫、数据分析的知识!