Python操作Excel:xlrd/xlwt读写指南与进阶技巧
本文详细介绍Python中利用xlrd和xlwt库进行Excel文件读写的方法。涵盖安装依赖、打开工作簿、获取Sheet信息、读写单元格数据及样式等核心功能,并深入讲解过滤隐藏Sheet、获取可见行列及单元格颜色样式等进阶技巧,帮助开发者高效实现办公自动化。
准备
在Python中处理Excel文件,主要有以下几种常用方案:
- xlrd / xlwt
- openpyxl
- Pandas
- xlsxwriter
- xlwings
- pywin32
其中,xlrd和xlwt是处理Excel文件最基础且广泛使用的两个第三方库。
具体分工上,
xlrd主要用于读取Excel文件,而xlwt则用于将数据写入Excel文件。
首先,我们需要安装这两个依赖库。
# 安装依赖库
pip3 install xlrd
pip3 install xlwt
xlrd读取Excel
利用xlrd库中的open_workbook(filepath)函数,可以打开本地的一个Excel文件。
import xlrd
# 打开文件,返回一个工作簿对象
wb = xlrd.open_workbook(file_path)
通过工作簿对象的nsheets属性可以获取Sheet的数量,使用sheet_names()方法则能返回所有Sheet名称的列表。
# 统计sheet数量
sheets_num, sheets_names = wb.nsheets, wb.sheet_names()
print('sheet数量一共有:', sheets_num)
print('sheet名称分别为:', sheets_names)
从工作簿中筛选特定Sheet有两种方式:
- 通过Sheet名称指定。
- 通过位置索引指定,索引从0开始。
# 获取某一个sheet
# 通过名称或者索引获取
sheet = wb.sheet_by_index(0)
# sheet = wb.sheet_by_name('第一个Sheet')
print(sheet)
每个Sheet对象都提供了name、nrows、ncols属性,分别用于获取Sheet名称、行数和列数。
此外,
row_values(index)和col_values(index)方法分别用于获取指定行或列的数据列表。
# 获取某一个sheet中,包含的行数量、列数量
sheet_name, sheet_row_count, sheet_column_count = sheet.name, sheet.nrows, sheet.ncols
print('当前sheet名称为:', sheet_name, ",一共有:", sheet_row_count, "行;有:", sheet_column_count, "列")
# 单独获取某一行数据,索引从0开始
# 比如:获取第2行数据
row_datas = sheet.row_values(1)
print('第2行数据为:', row_datas)
# 单独获取某一列数据,索引从0开始
# 比如:获取第二列数据
column_datas = sheet.col_values(1)
print('第2列数据为:', column_datas)
若需获取特定单元格数据,可通过行索引和列索引调用cell(row_index, column_index)函数。
需注意,行索引和列索引均从0开始,即0代表第一行。
在xlrd中,单元格数据类型分为6种,通过ctype属性对应如下:
- 0 -- 空(empty)
- 1 -- 字符串(string)
- 2 -- 数字(number)
- 3 -- date(日期)
- 4 -- boolean(布尔值)
- 5 -- error(错误)
# 获取某一个单元格的数据
# 比如:获取第2行第1列的单元格的数据
one_cell = sheet.cell(1, 0)
# 单元格的值
cell_value = one_cell.value
print("单元格的值为:", cell_value)
# 单元格数据类型
cell_type = one_cell.
print("单元格数据类型为:", cell_type)
最后,若要获取当前Sheet所有单元格的数据,可以通过遍历所有行和列来实现。
# 获取所有单元格的值
print('表格中所有数据如下:')
for r in range(sheet.nrows):
for i in range(sheet.ncols):
print(sheet.cell(r, i).value)
xlwt写入Excel
若需将数据写入Excel,使用xlwt库非常便捷。
首先,调用xlwt的Workbook()方法创建一个工作簿对象。
接着,利用工作簿对象的add_sheet(sheetname)方法新增一个Sheet。
import xlwt
sheetname = '第一个Sheet'
# 创建一个工作簿对象
wb = xlwt.Workbook()
# 添加Sheet,通过sheet名称
sheet = wb.add_sheet(sheetname)
随后,通过sheet对象的write()方法,依据行索引和列索引将数据写入对应单元格。
# 将数据写入到Sheet中
# 3个参数分别是:行索引(从0开始)、列索引(从0开始)、单元格的值
# 第一行第一列,写入一个数据
# 写入标题
for index, title in enumerate(self.titles):
sheet.write(0, index, title)
# 写入值
for index_row, row_values in enumerate(self.values):
for index_column, column_value in enumerate(row_values):
sheet.write(index_row + 1, index_column, column_value)
需要注意的是,最后必须调用工作簿的save(filepath)方法,才能在本地生成Excel文件。
# 保存文件
# 最后保存文件即可
wb.save(filepath)
进阶用法
接下来介绍几个常用的进阶操作。
1、获取所有可见的Sheet
在读取Sheet数据时,通常需要过滤掉隐藏的Sheet。
当sheet对象的visibility属性值为0时,表示该Sheet在工作簿中可见;否则表示被隐藏。
def get_all_visiable_sheets(self, wb):
"""
获取所有可见的sheet
:param wb:
:return:
"""
return list(filter(lambda item: item.visibility == 0, wb.sheets()))
# 1、获取所有可看见的sheet
sheet_visiable = self.get_all_visiable_sheets(wb)
print('所有可见的sheet包含:', sheet_visiable)
2、获取Sheet可见行或列
某个Sheet中可能存在部分行或列被隐藏的情况。
def get_all_visiable_rows(self, sheet):
"""
获取某一个sheet中,可见的行
:param sheet:
:return:
"""
result = [index for index in range(sheet.nrows) if sheet.rowinfo_map[index].hidden == 0]
return result
def get_all_visiable_columns(self, sheet):
"""
获取某一个sheet中,可见的列
:param sheet:
:return:
"""
result = [index for index in range(sheet.ncols) if sheet.colinfo_map[index].hidden == 0]
return result
3、获取单元格的样式
以获取单元格字体颜色和背景色为例。
def get_cell_bg_color(self, wb, sheet, row_index, col_index):
"""
获取某一个单元格的背景颜色
:param wb:
:param sheet:
:param row_index:
:param col_index:
:return:
"""
xfx = sheet.cell_xf_index(row_index, col_index)
xf = wb.xf_list[xfx]
# 字体颜色
font_color = wb.font_list[xf.font_index].colour_index
# 背景颜色
bg_color = xf.background.pattern_colour_index
return font_color, bg_color
需注意,使用xlrd读取单元格样式时,打开工作簿需显式定义formatting_info = True,否则会抛出异常。
# 注意:必须设置formatting_info=True,才能正常获取属性
wb = xlrd.open_workbook(file_path, formatting_info=True)
sheet = wb.sheet_by_index(0)
最后
搭配使用xlrd和xlwt,基本能完成大部分常规工作。对于复制、分割、筛选等复杂功能,可借助xlutils库。
需指出的是,该组合对xlsx格式的兼容性较差;若需操作xlsx文件,需先将其转换为xls格式后再进行处理。
代码地址:https://github.com/xingag/test_auto/tree/master/office_auto/Excel
以上就是python办公自动化之excel的操作的详细内容,更多关于python excel自动化的资料请关注本站其它相关文章!
免责声明:本文内容仅供参考,不代表本站观点。转载或引用请注明出处。









