一、需求描述:
1、一张人员信息表中生成人员信息,某些列的字段要进行递增操作;
2、一个组织节点下存在1000人的限制要求;
3、一张Excel表格生成45000条数据;
二、Excel表格的表头如下:
三、相关操作
1、写Excel,首先需要安装xlwt模块;
2、可以采用pip的安装方式,安装方式如下图所示:
3、安装完成后,开始编写我们的程序:
首先需要明确几个函数的用法:
1、enumerate()
详细介绍:
2、format()
格式化数据
3、yield相关知识
详细介绍:http://blog.csdn.net/alvine008/article/details/43410079
写数据基本操作:
创建工作簿(workbook)和工作表(sheet):
import xlwt
workbook = xlwt.Workbook()
sheet = workbook.add_sheet("Sheet Name")
写单元格(cell):
sheet.write(0, 0, ‘foobar‘) # row, column, value
对单元格应用样式(加粗为例):
style = xlwt.easyxf(‘font: bold 1‘)
sheet.write(0, 0, ‘foobar‘, style)
具体实现代码:
#! -*- coding:utf-8 -*-
‘‘‘
1、第一步:我们需要引入xlwt模块:import xlwt
2、第二步:创建一个excel表格
3、第三步:定义需要自增字段的起始值
4、第四步:初始化表格:sheet页、样式、表头填充
5、第五步:按照每行进行数据写入,1个部门节点人数限制为1000人
‘‘‘
import xlwt
EXCEL_NAME = ‘employees_mock.xls‘
PHONE_NUM = 13805710001
EMP_NUM = 482501
EMAIL_PREFIX = 333019041
SUB_PHONE_NUM = 478180
OFFICE_SUFFIX = 12771500
VIRTUAL_NET = 123456
TELGRAM_PHONE_NUM = 15370032950
def employe(depart, count):
for _ in range(1000):
name = u‘工作人员{}‘.format(count)
phone_num = str(PHONE_NUM + count)
gender = u‘男‘
emp_num = str(EMP_NUM + count)
is_charger = u‘否‘
job = u‘开发‘
email = u‘{}@foo.com‘.format(EMAIL_PREFIX + count)
sub_phone_num = SUB_PHONE_NUM + count
addr = u‘杭州滨江‘
remark = u‘测试‘
office = u‘信息安全管理员‘
position = u‘管理岗位‘
bureau = u‘发发发‘
office_phone_num = u‘0571-{}‘.format(OFFICE_SUFFIX)
v_net = u‘192.168.1.{}‘.format(count % 256)
tel_phone_num = str(TELGRAM_PHONE_NUM + count)
tel_net = u‘192.168.0.{}‘.format(count % 256)
yield [
depart,
name,
phone_num,
gender,
emp_num,
is_charger,
job,
email,
sub_phone_num,
addr,
remark,
office,
position,
bureau,
office_phone_num,
v_net,
tel_phone_num,
tel_net
]
count += 1
def style():
##赋值style为XFStyle(),初始化样式
style = xlwt.XFStyle()
#设置单元格内字体样式
font = xlwt.Font()
font.name = ‘Times New Roman‘
font.bold = False
return style
#写Excel
def write_excel():
wb = xlwt.Workbook()#创建工作簿
sheet = wb.add_sheet(u‘sheet1‘, cell_overwrite_ok=True)#创建第一个sheet页 第二参数用于确认同一个cell单元是否可以重设值
#初始化表头列表
tb_head = [
u‘部门‘,
u‘姓名‘,
u‘手机号‘,
u‘性别‘,
u‘工号‘,
u‘是否此部门主管(是/否)‘,
u‘职位‘,
u‘邮箱‘,
u‘分机号‘,
u‘办公地点‘,
u‘备注‘,
u‘职务‘,
u‘岗位‘,
u‘办公室‘,
u‘办公电话‘,
u‘虚拟网‘,
u‘电信手机号‘,
u‘电信虚拟网‘
]
for i, item in enumerate(tb_head):
sheet.write(0, i, item, style())
return wb
def main():
wb = write_excel()
table = wb.get_sheet(0)
# insert 10k mock employee data
deps = [u‘浙江省-反贪局-分局{}‘.format(i + 1) for i in range(10)]
# deps.extend(
# [u‘浙江省-工商局-分局{}‘.format(i) for i in range(10)]
# )
# deps.extend(
# [u‘陕西省-计划生育局-分局{}‘.format(i) for i in range(10)]
# )
for c, dep in enumerate(deps):
count = 1000 * c + 1
for row, emp in enumerate(employe(dep, count)):
for col, item in enumerate(emp):
table.write(row + count, col, item)
#保存文件
wb.save(EXCEL_NAME)
if __name__ == ‘__main__‘:
main()