标签:Python与Excel,xlwt
有时候,不需要调用Excel来处理电子表格数据。例如,使用xlwt。
首先,使用pip命令在终端安装xlwt:
pip install xlwt
下面是一个示例。原始的文本文件数据如下:
09700RESEARCH
09800PHYSICIANS PRIVATE OFFICES
09900NONPAID WORKERS
MANAGEMENT FEES
REFERENCE LABS
原始数据被搅和在一起,账号和类别没有分开,有些数据甚至没有账号。将这些数据写入Excel工作表并格式化,如下图1所示。
图1
要创建这样的输出,代码脚本执行以下操作:
1.分隔帐号和名称
2.分配一个99999的帐号,并将未编号帐号的单元格颜色设置为红色
3.将帐户名转换为正确的大写名称
4.删除帐户名中的任何多余空格
5.将账号和姓名写入电子表格中的两列
6.根据最宽数据的宽度设置每个电子表格列的列宽格式
代码如下:
import sys
import re
from xlwt import Workbook, easyxf
def dox1():
try:
fp = open(r"C:\test\hospdata.txt")
except:
print('打开hospdata.txt失败')
sys.exit(1)
lines = fp.readlines()
nameandnum = re.compile(r'(\d+)\s*(.*)\s*')
wb = Workbook()
wsraw = wb.add_sheet('Raw Data')
ws = wb.add_sheet('Account List')
ws.write(0, 0, 'Account Number')
ws.write(0, 1, 'Account Name')
ws.col(0).width = len('Account Number') * 256
ws.col(1).width = max([len(line) for line in lines]) * 256
r = 1
for line in lines:
wsraw.write(r, 0, line.strip())
m = nameandnum.match(line)
if m:
ws.write(r, 0, int(m.group(1)))
ws.write(r, 1, ' '.join([w.capitalize() for w in m.group(2).split()]))
else:
ws.write(r, 0, 99999, easyxf('pattern: pattern solid, fore_colour red;'))
ws.write(r, 1, ' '.join([w.capitalize() for w in line.split()]))
r += 1
wb.save(r'C:\test\accounts.xls')
print('已写入accounts.xls')
if __name__ == "__main__":
dox1()
注:本文学习整理自pythonexcels.com,供参考。