爬虫与Python：（四）爬虫进阶二之数据存储（文件存储）——4.Excel文件存储

实现对Excel文件的操作，需要引入第三方模块——xlwt、xlwr。

• wlwt：实现对Excel文件的写入。
• xlwr:实现对Excel文件内容的读取。

写入Excel文件示例代码如下：

import xlwt

# 写入Excel文件
'''
设置格式
@:param name 字体
@:param height 高度
@:param bold 是否加粗
'''
def set_style(name, height , bold=False):
    style = xlwt.XFStyle()
    font = xlwt.Font()
    font.name = name
    font.bold = bold
    font.colour_index = 4
    font.height = height
    style.font = font
    return  style

'''
写入excel表格
@:param path 文件路径
'''
def write_excel(path):
    workbook = xlwt.Workbook(encoding='utf-8')
    data_sheet = workbook.add_sheet('demo') # 创建工作簿
    row0=[u'字段名称',u'大致时段','CRNT','CELL-ID']
    row1=[u'测试','15:50:33-15:52:14',22706,4190202]
    # 循环为单元格写入数据
    for i in range(len(row0)):
        data_sheet.write(0,i ,row0[i],set_style("Times New Roman",220 ,True))   # 写操作
        data_sheet.write(1,i ,row1[i],set_style("Times New Roman",220 ,True))
    workbook.save(path)     # 保存工作簿


if __name__ == '__main__':
    path = 'demo.xls'
    write_excel(path)       # 写入Excel
    print('创建demo.xls文件成功')

运行后生成了一个Excel文件，内容如图1所示。

 图1

关键代码如下：

workbook = xlwt.Workbook(encoding='utf-8')  # 实例化Workbook
data_sheet = workbook.add_sheet('demo') # 创建sheet
workbook.save(path)  # 保存文件
data_sheet.write(行,列,value)  # 写操作

使用xlrd模块对Excel文件进行读取，示例代码如下：

# 对Excel文件进行读操作
import xlrd

Workbook = xlrd.open_workbook('demo.xls')

sheet_names = Workbook.sheet_names()
sheet1 = Workbook.sheet_by_name('demo')
sheet1 = Workbook.sheet_by_index(0)
rows = sheet1.row_values(1)
cols10 = sheet1.col_values(1)
print('rows',rows)
print('cols10',cols10)

参数说明如下：

1.  xlrd.open_workbook() ：表示打开Excel文件。
2.  Workbook.sheet_by_name('demo') ：表示获取所在的sheet表数据。
3.  Workbook.sheet_by_index(0) ：表示获取第一张sheet表名称，根据索引来取值，从0开始。
4.  rows = sheet1.row_values(1) ：表示获取sheet1中第2行数据。
5.  cols10 = sheet1.col_values(1) ：表示获取sheet1中第二列数据。