python实现xlwt xlrd 指定条件给excel行添加颜色

Z时代
2024-01-10
分类：综合

先用xlrd读excel文件--》book对象a

拿到指定的sheet页 xlrd对象

用xlutils copy 的copy方法复制 a得到b

通过判断a的列值，来修改b

保存b 得到结果

之前也有试过直接用xlwt 来操作sheetwork对象，来实现给指定行添加颜色的操作，由于能力有限，最终没有找到合适的方法，最终换了个方法先读出来，因为workbook对象可以拿到行数和对列操作，筛选关键字比较方便，所以上边代码就是一个demo，但是这个方法还是有弊端的，我把dataframe导出到excel，没有合适的方法给指定列添加颜色，只能先临时落到本地，再读出来，很麻烦，不知道有没有sheetwork 直接转workbook的方法，还得再学习啊~

补充知识：python 如何对excel中某一列某些值的单元格着色

效果：

代码如下：


import xlwt
import xlrd
from xlutils.copy import copy
#创建execl
def create_execl(file_name):
 wb = xlwt.Workbook()
 ws = wb.add_sheet('Info')
 ws.write(0, 0, "1")
 ws.write(1, 0, "2")
 ws.write(2, 0, "3")
 ws.write(3, 0, "2")
 wb.save(file_name)
#单元格上色
def color_execl(file_name):
 styleBlueBkg = xlwt.easyxf('pattern: pattern solid, fore_colour red;') # 红色
 rb = xlrd.open_workbook(file_name)  #打开t.xls文件
 ro = rb.sheets()[0]      #读取表单0
 wb = copy(rb)       #利用xlutils.copy下的copy函数复制
 ws = wb.get_sheet(0)     #获取表单0
 col = 0         #指定修改的列
 for i in range(ro.nrows):    #循环所有的行
  result = int(ro.cell(i, col).value)
  if result == 2:      #判断是否等于2
   ws.write(i,col,ro.cell(i, col).value,styleBlueBkg)
 wb.save(file_name)
if __name__ == '__main__':
 file_name = 't.xls'
 create_execl(file_name)
 color_execl(file_name)