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

Python笔记（十四）：操作excel openpyxl模块

2018-06-18 03:24:38来源：未知阅读 ()

（一） 常遇到的情况

就我自己来说，常遇到的情况可能就下面几种：

读取excel整个sheet页的数据。
读取指定行、列的数据
往一个空白的excel文档写数据
往一个已经有数据的excel文档追加数据

下面就以这几种情况为例进行说明。

（二） 涉及的模块及函数说明

就我知道的，有3个模块可以操作excel文档，3个模块通过pip都可以直接安装。

xlrd:读数据

xlwt:写数据

openpyxl：可以读数据，也可以写数据

这里就就只说明openpyxl了，因为这个模块能满足上面的需要了。

openpyxl函数

函数	说明
load_workbook(filename)	打开excel，并返回所有sheet页访问指定sheet页的方法： #打开excel文档 wb = openpyxl.load_workbook(file_name) #访问sheet页 sheet = wb[‘sheet页的名称’] #关闭excel文档 wb.close()
Workbook()	创建excel文档 wb = openpyxl.Workbook() #保存excel文档 wb.save（'文件名.xlsx'）
下面的函数是针对sheet页的 sheet = wb[‘sheet页的名称’] 访问指定单元格的方式sheet['A1']、sheet['B1']...
min_row	返回包含数据的最小行索引，索引从1开始例如：sheet.min_row
max_row	返回包含数据的最大行索引，索引从1开始
min_column	返回包含数据的最小列索引，索引从1开始
max_column	返回包含数据的最大列索引，索引从1开始
values	获取excel文档所有的数据，返回的是一个generator对象
iter_rows(min_row=None, max_row=None, min_col=None, max_col=None)	min_row:最小行索引 max_row:最大行索引 min_col:最小列索引 max_col:最大列索引获取指定行、列的单元格，没指定就是获取所有的

现在我有这么一个excel，下面以这个excel进行说明。

关于min_row、max_row这些，看下面的输出就很直观了

 1 import openpyxl
 2 
 3 def get_data_openpyxl(file_name,sheet):
 4     #打开excel文档
 5     wb = openpyxl.load_workbook(file_name)
 6     #访问sheet页
 7     sheet = wb[sheet]
 8     # 包含数据的最小行索引，从1开始
 9     minRow = sheet.min_row
10     print("最小行索引是：", minRow)
11     #包含数据的最大行索引，从1开始
12     maxRow = sheet.max_row
13     print("最大行索引是：",maxRow)
14     #包含数据的最小列索引
15     minColumn = sheet.min_column
16     print("最小列索引是：",minColumn)
17     #包含数据的最大列索引
18     maxColumn = sheet.max_column
19     print("最大列索引是：", maxColumn)
20     wb.close()
21 get_data_openpyxl('测试.xlsx','Sheet')

（三） 读取excel整个sheet页的数据

下面的代码都是没加异常处理的，要加的话自己看情况加上异常处理就行了。

 1 import openpyxl
 2 
 3 def get_data_openpyxl(file_name,sheet):
 4     #打开excel文档
 5     wb = openpyxl.load_workbook(file_name)
 6     #访问sheet页
 7     sheet = wb[sheet]
 8     # 获取excel文档所有的数据,返回的是一个generator对象
 9     data = sheet.values
10     print(data)
11     #迭代输出所有数据
12     for i in data:
13         print(i)
14     wb.close()
15 get_data_openpyxl('测试.xlsx','Sheet')

（四） 读取指定行、列的数据

这里有个问题就是，openpyxl模块貌似没有读取指定行、列数据的函数，不过没关系，自己封装一个函数去实现就行了，这个是通用的（前提是已经安装openpyxl），可以创建一个类（可以根据函数的作用创建多个不同的类，这个看自己了），放一些自己写的常用函数。

 1 import openpyxl
 2 
 3 def get_data_iter(file_name,sheet, max_row=None,min_row=None,max_col=None,min_col=None):
 4     '''
 5 
 6     :param file_name: excel文件名称
 7     :param sheet: sheet页名称
 8     :param max_row:最大行索引，未指定则获取所有行的数据
 9     :param min_row: 最小行索引，未指定则从第一行开始
10     :param min_col:最小列索引，未指定则从第一列开始
11     :param max_col:最大列索引，未指定则获取所有列的数据
12     :return:返回指定行、列的数据
13     '''
14     # 打开excel文档
15     wb = openpyxl.load_workbook(file_name)
16     # 访问sheet页
17     sheet = wb[sheet]
18     # 获得指定行列的单元格
19     cell = sheet.iter_rows(max_row=max_row, min_row=min_row, max_col=max_col, min_col=min_col)
20     all_rows = []
21     # 获取单元格的值
22     for row in cell:
23         rows = []
24         for c in row:
25             rows.append(c.value)
26         all_rows.append(tuple(rows))
27     wb.close()
28     return all_rows
29 
30 rows = get_data_iter('测试.xlsx','Sheet',max_row=10,min_row=5,max_col=3,min_col=1)
31 print(rows)
32 for i in rows:
33     print(i)

（五） 往空白的excel文档写数据

 1 import openpyxl
 2 
 3 #创建excel文档
 4 wb =openpyxl.Workbook()
 5 sheet = wb['Sheet']
 6 sheet['A1'] = '颜色'
 7 sheet['B1'] = '版本'
 8 x = 2
 9 for i in range(10):
10     sheet['A'+str(x)] = i+1
11     sheet['B'+str(x)] = i+1
12     x += 1
13 
14 wb.save('测试写数据.xlsx')

执行后，可以在当前工作目录下看到这个excel文档

（六）往一个已经有数据的excel文档追加数据

要追加数据的话，获取已经有数据的最大索引就行了，从下一行开始添加数据，这里X的初始值忘记加1了，代码就不修改了，能看明白就行了

 1 import openpyxl
 2 
 3 # 打开excel文档
 4 wb = openpyxl.load_workbook('测试写数据.xlsx')
 5 # 访问sheet页
 6 sheet = wb['Sheet']
 7 #获取最大行索引
 8 maxRow = sheet.max_row
 9 x = maxRow
10 for i in range(10):
11     sheet['A'+str(x)] = '追加数据'
12     sheet['B'+str(x)] = '追加数据'
13     x += 1
14 
15 wb.save('测试写数据.xlsx')