欢迎访问宙启技术站
智能推送

Python中使用Workbook()函数提取Excel文件中的特定数据

发布时间:2023-12-24 01:49:05

在Python中,可以使用openpyxl库来操作Excel文件。openpyxl库提供了一种易于使用的方式来读取和写入Excel文件,并且支持提取特定数据。

首先,需要安装openpyxl库,可以使用以下命令来进行安装:

pip install openpyxl

接下来,可以使用Workbook()函数创建一个新的Excel文件。以下是一个示例代码:

from openpyxl import Workbook

# 创建一个新的Excel文件
workbook = Workbook()

# 获取工作簿的活动表
sheet = workbook.active

# 设置表头
sheet["A1"] = "姓名"
sheet["B1"] = "年龄"
sheet["C1"] = "性别"

# 写入数据
sheet["A2"] = "张三"
sheet["B2"] = 20
sheet["C2"] = "男"
sheet["A3"] = "李四"
sheet["B3"] = 25
sheet["C3"] = "女"

# 保存Excel文件
workbook.save("data.xlsx")

上述代码创建了一个新的Excel文件,并在第一行中设置了表头,然后在第二行和第三行写入了数据。

要从现有的Excel文件中提取特定的数据,可以使用load_workbook()函数来加载文件,并使用sheet属性来获取特定的表。

以下是一个示例代码:

from openpyxl import load_workbook

# 加载Excel文件
workbook = load_workbook(filename="data.xlsx")

# 获取工作簿的活动表
sheet = workbook.active

# 获取特定单元格的数据
name = sheet["A2"].value
age = sheet["B2"].value
gender = sheet["C2"].value

# 打印提取的数据
print("姓名:", name)
print("年龄:", age)
print("性别:", gender)

上述代码加载了之前保存的Excel文件,并从第二行中提取了姓名、年龄和性别的数据,并打印出来。

除了使用单元格的坐标来提取数据之外,还可以使用iter_rows()方法来迭代行,并使用每一行中的单元格的值来提取特定数据。

以下是一个示例代码:

from openpyxl import load_workbook

# 加载Excel文件
workbook = load_workbook(filename="data.xlsx")

# 获取工作簿的活动表
sheet = workbook.active

# 迭代行
for row in sheet.iter_rows(min_row=2, max_row=3):
    name = row[0].value
    age = row[1].value
    gender = row[2].value
    print("姓名:", name)
    print("年龄:", age)
    print("性别:", gender)

上述代码从第二行到第三行迭代行,并提取每一行中的姓名、年龄和性别的数据,并打印出来。

使用Workbook()函数提取Excel文件中的特定数据就是这样。通过openpyxl库的使用,可以方便地操作Excel文件,并提取所需的数据。