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文件,并提取所需的数据。
