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

（一）  常遇到的情况

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

1. 读取excel整个sheet页的数据。
2. 读取指定行、列的数据
3. 往一个空白的excel文档写数据
4. 往一个已经有数据的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这些，看下面的输出就很直观了

import openpyxl

def get_data_openpyxl(file_name,sheet):
    #打开excel文档
    wb = openpyxl.load_workbook(file_name)
    #访问sheet页
    sheet = wb[sheet]
    # 包含数据的最小行索引，从1开始
    minRow = sheet.min_row
    print("最小行索引是：", minRow)
    #包含数据的最大行索引，从1开始
    maxRow = sheet.max_row
    print("最大行索引是：",maxRow)
    #包含数据的最小列索引
    minColumn = sheet.min_column
    print("最小列索引是：",minColumn)
    #包含数据的最大列索引
    maxColumn = sheet.max_column
    print("最大列索引是：", maxColumn)
    wb.close()
get_data_openpyxl('测试.xlsx','Sheet')

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

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

import openpyxl

def get_data_openpyxl(file_name,sheet):
    #打开excel文档
    wb = openpyxl.load_workbook(file_name)
    #访问sheet页
    sheet = wb[sheet]
    # 获取excel文档所有的数据,返回的是一个generator对象
    data = sheet.values
    print(data)
    #迭代输出所有数据
    for i in data:
        print(i)
    wb.close()
get_data_openpyxl('测试.xlsx','Sheet')

 

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

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

import openpyxl

def get_data_iter(file_name,sheet, max_row=None,min_row=None,max_col=None,min_col=None):
    '''

    :param file_name: excel文件名称
    :param sheet: sheet页名称
    :param max_row:最大行索引，未指定则获取所有行的数据
    :param min_row: 最小行索引，未指定则从第一行开始
    :param min_col:最小列索引，未指定则从第一列开始
    :param max_col:最大列索引，未指定则获取所有列的数据
    :return:返回指定行、列的数据
    '''
    # 打开excel文档
    wb = openpyxl.load_workbook(file_name)
    # 访问sheet页
    sheet = wb[sheet]
    # 获得指定行列的单元格
    cell = sheet.iter_rows(max_row=max_row, min_row=min_row, max_col=max_col, min_col=min_col)
    all_rows = []
    # 获取单元格的值
    for row in cell:
        rows = []
        for c in row:
            rows.append(c.value)
        all_rows.append(tuple(rows))
    wb.close()
    return all_rows

rows = get_data_iter('测试.xlsx','Sheet',max_row=10,min_row=5,max_col=3,min_col=1)
print(rows)
for i in rows:
    print(i)

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

import openpyxl

#创建excel文档
wb =openpyxl.Workbook()
sheet = wb['Sheet']
sheet['A1'] = '颜色'
sheet['B1'] = '版本'
x = 2
for i in range(10):
    sheet['A'+str(x)] = i+1
    sheet['B'+str(x)] = i+1
    x += 1

wb.save('测试写数据.xlsx')

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

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

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

import openpyxl

# 打开excel文档
wb = openpyxl.load_workbook('测试写数据.xlsx')
# 访问sheet页
sheet = wb['Sheet']
#获取最大行索引
maxRow = sheet.max_row
x = maxRow
for i in range(10):
    sheet['A'+str(x)] = '追加数据'
    sheet['B'+str(x)] = '追加数据'
    x += 1

wb.save('测试写数据.xlsx')

执行完后：