Python操作Excel表格

转载请注明出处❤️

作者:测试蔡坨坨

原文链接:caituotuo.top/b0277e1c.html


你好,我是测试蔡坨坨。

上期我们分享了一个Python编写的小工具——「Python 实现 XMind 测试用例快速转 Excel 用例

其中用到了Python操作Excel,有小伙伴可能对这部分不太了解,所以本篇将介绍Python是如何将数据写入Excel表格。

安装

xlwt是Python的第三方模块,需要先下载安装才能使用,这里我们使用pip命令下载

1
pip3 install xlwt

使用

  1. 导入Excel表格文件处理函数

    1
    2
    3
    import xlrd
    import xlwt
    from faker import Faker
  2. 创建Excel表格类型文件

    1
    2
    3
    4
    # 实例化Workbook对象
    # 第一个参数:encoding表示编码
    # 第二个参数:style_compression设置是否压缩,0表示不压缩
    work_book = xlwt.Workbook(encoding="utf-8", style_compression=0)
  3. 在Excel表格类型文件中建立一张表sheet表单

    1
    2
    3
    # 第一个参数:sheetname,表示sheet名
    # 第二个参数:cell_overwrite_ok用于确认同一cell单元是否可以重设值,True表示可以重设
    sheet = work_book.add_sheet(sheetname="用户信息表", cell_overwrite_ok=True)
  4. 自定义列名

    1
    2
    # 用一个元组col自定义列的数量以及属性
    col = ("姓名", "电话", "地址")
  5. 将列属性元组col写进sheet表单中

    1
    2
    3
    4
    # 使用for循环将col元组的元组值写到sheet表单中
    # 第一个参数是行,第二个参数是列,第三个参数是值
    for i in range(0, 3):
    sheet.write(0, i, col[i])
  6. 创建数据并将数据写入表格

    1
    2
    3
    4
    5
    6
    7
    8
    9
    10
    11
    12
    13
    14
    15
    # 使用Faker模块生成10组数据
    faker = Faker("zh_CN")
    data_list = []
    for i in range(0, 10):
    data = [faker.name(), faker.phone_number(), faker.address()]
    data_list.append(data)
    print(data_list) # [['杨雪梅', '13596272521', '湖南省宁德市高明杨街Z座 257668'], ……]

    # 将数据写入Excel文件
    # 先用第一个for循环进行每行写入
    # 再用第二个for循环把每一行当中的列值写进入
    for i in range(0, 10):
    data = data_list[i]
    for j in range(0, 3):
    sheet.write(i + 1, j, data[j])
  7. 保存Excel文件,调用save()方法

    1
    2
    3
    # 定义一个文件路径save_path,例如当前目录下./ 文件名为 userinfo.xls
    save_path = "./userinfo.xls"
    work_book.save(save_path)
  8. 读取Excel文件(ps:读取前确保文件非打开状态)

    1
    2
    3
    4
    5
    6
    7
    8
    9
    10
    11
    12
    13
    # 得到文件
    file_name = xlrd.open_workbook("./userinfo.xls")
    # 得到sheet页
    sheet = file_name.sheets()[0]
    # 获取总列数
    total_rows = sheet.nrows
    # 获取总行数
    total_cols = sheet.ncols
    print(total_rows, total_cols) # 11 3
    for i in range(1, total_rows):
    for j in range(0, total_cols):
    info = sheet.row_values(i)[j]
    print(info)

运行结果:借助Faker模块生成随机的个人信息,并将其写入Excel表格

完整代码

源码获取请关注公众号测试蔡坨坨,回复关键词源码

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
# author: 测试蔡坨坨
# datetime: 2022/7/2 20:47
# function: Python操作Excel表格

# xlwt是Python的第三方模块,需要先下载安装才能使用,这里我们使用pip命令下载
# pip3 install xlwt

# 1.导入Excel表格文件处理函数
import xlrd
import xlwt
from faker import Faker

# 2.创建Excel表格类型文件
# 实例化Workbook对象
# 第一个参数:encoding表示编码
# 第二个参数:style_compression设置是否压缩,0表示不压缩
work_book = xlwt.Workbook(encoding="utf-8", style_compression=0)

# 3.在Excel表格类型文件中建立一张表sheet表单
# 第一个参数:sheetname,表示sheet名
# 第二个参数:cell_overwrite_ok用于确认同一cell单元是否可以重设值,True表示可以重设
sheet = work_book.add_sheet(sheetname="用户信息表", cell_overwrite_ok=True)

# 4.自定义列名
# 用一个元组col自定义列的数量以及属性
col = ("姓名", "电话", "地址")

# 5.将列属性元组col写进sheet表单中
# 使用for循环将col元组的元组值写到sheet表单中
# 第一个参数是行,第二个参数是列,第三个参数是值
for i in range(0, 3):
sheet.write(0, i, col[i])

# 6.创建数据并将数据写入表格
# 使用Faker模块生成10组数据
faker = Faker("zh_CN")
data_list = []
for i in range(0, 10):
data = [faker.name(), faker.phone_number(), faker.address()]
data_list.append(data)
print(data_list) # [['杨雪梅', '13596272521', '湖南省宁德市高明杨街Z座 257668'], ……]

# 将数据写入Excel文件
# 先用第一个for循环进行每行写入
# 再用第二个for循环把每一行当中的列值写进入
for i in range(0, 10):
data = data_list[i]
for j in range(0, 3):
sheet.write(i + 1, j, data[j])

# 7.保存Excel文件,调用save()方法
# 定义一个文件路径save_path,例如当前目录下./ 文件名为 userinfo.xls
save_path = "./userinfo.xls"
work_book.save(save_path)

# 8.读取Excel文件(ps:读取前确保文件非打开状态)
# 得到文件
file_name = xlrd.open_workbook("./userinfo.xls")
# 得到sheet页
sheet = file_name.sheets()[0]
# 获取总列数
total_rows = sheet.nrows
# 获取总行数
total_cols = sheet.ncols
print(total_rows, total_cols) # 11 3
for i in range(1, total_rows):
for j in range(0, total_cols):
info = sheet.row_values(i)[j]
print(info)