在利用Python处理Excel文件的时候,经常会遇到要读取某一个单元格的数据,经过程序处理得到新数据,想把新数据插入到同一个Excel的下一列的任务。不想新建一个Excel表格,只想在同一文件上进行数据存取插入。
这时可以利用Python提供的Excel工具包模块xlwt, xlrd, xlutils实现上述功能。
import xlwt
import xlrd
from xlutils.copy import copy
class ExcelProcess:
def __init__(self, doc_file, sheet_name):
self.doc_file = doc_file
self.sheet_name = sheet_name
def get_rows_number(self):
"""
获取某一table中数据的总行数
:return: 数据的总行数
"""
data = xlrd.open_workbook(self.doc_file)
table = data.sheet_by_name(self.sheet_name)
return table.nrows
def excel_read(self, x, y):
"""
读取Excel工作表中某一table的单元格的值
:param x:
:param y:
:return: 单元格的值
"""
data = xlrd.open_workbook(self.doc_file)
table = data.sheet_by_name(self.sheet_name)
return table.cell(x, y).value
def excel_create(self, x, y, value):
"""
创建一个Excel工作表
:param x: 单元格横坐标
:param y: 单元格纵坐标
:param value: 要写入单元格的值
:return:
"""
data = xlwt.Workbook()
table = data.add_sheet(self.sheet_name)
table.write(x, y, value)
data.save(self.doc_file)
def excel_change(self, x, y, value):
"""
修改原始excel文件
:param x: 单元格横坐标
:param y: 单元格纵坐标
:param value: 要写入单元格的值
:return:
"""
rb = xlrd.open_workbook(self.doc_file)
# 管道作用
wb = copy(rb)
# 通过get_sheet()获取的sheet有write()方法
# ws = wb.get_sheet(0) # 1代表是写到第几个工作表里,从0开始算是第一个。
ws = wb.sheet_by_name(self.sheet_name)
ws.write(x, y, value)
wb.save(self.doc_file)
写了一个Excel处理的工具类,用于实现Excel工作表的创建(excel_create函数),Excel工作表的读取(excel_read函数),Excel工作表的修改(excel_change函数)和获取Excel工作表的数据行数(get_rows_number函数)功能。
其中,excel_change函数主要是利用一个管道来实现同时读写的任务,利用了xlutils工具包中提供的copy功能,再把copy的表写入新数据覆盖原有的工作表。