使用openpyxl库从Excel文件中提取指定的数据并生成新的文件

简介: 使用openpyxl库从Excel文件中提取指定的数据并生成新的文件

一、引言

Excel文件作为数据处理和存储的重要工具,在日常办公和数据分析中扮演着不可或缺的角色。openpyxl是一个用于读取、写入Excel 2010 xlsx/xlsm/xltx/xltm文件的Python库。它允许用户通过编程的方式操作Excel文件,包括读取数据、修改数据以及生成新的Excel文件。本文将详细介绍如何使用openpyxl库从Excel文件中提取指定的数据,并生成新的Excel文件。通过案例和代码,帮助新手朋友快速掌握这一技能。

二、openpyxl库的安装与基本用法

在开始使用openpyxl之前,需要先安装这个库。可以通过pip命令在终端或命令提示符中安装:

pip install openpyxl

安装完成后,我们就可以开始使用openpyxl库来操作Excel文件了。下面是一个简单的示例,展示如何打开一个Excel文件并读取其中的数据:

1. from openpyxl import load_workbook  
2.   
3. # 加载Excel文件  
4. workbook = load_workbook('example.xlsx')  
5.   
6. # 获取活动工作表(默认打开的工作表)  
7. sheet = workbook.active  
8.   
9. # 读取单元格数据  
10. cell_value = sheet['A1'].value  
11. print(cell_value)

在上面的代码中,我们首先导入了openpyxl库中的load_workbook函数。然后,使用该函数加载了一个名为example.xlsx的Excel文件,并获取了其中的活动工作表。最后,通过指定单元格的位置(如'A1'),读取了该单元格中的数据并打印出来。

三、从Excel文件中提取指定数据

在实际应用中,我们通常需要提取Excel文件中特定区域或满足某些条件的数据。下面将介绍几种常见的提取数据的方法。

按行或列提取数据

我们可以使用openpyxl提供的迭代功能,按行或列遍历工作表中的数据。以下是一个按行提取数据的示例:

1. from openpyxl import load_workbook  
2.   
3. # 加载Excel文件  
4. workbook = load_workbook('example.xlsx')  
5. sheet = workbook['Sheet1']  # 指定工作表名称  
6.   
7. # 按行提取数据  
8. for row in sheet.iter_rows(min_row=2, max_col=3, max_row=5):  # 提取第2行到第5行,第1列到第3列的数据  
9.     for cell in row:  
10.         print(cell.value)

在上面的代码中,我们使用iter_rows方法按行遍历了指定区域的数据。通过指定min_row、max_col和max_row参数,我们可以控制提取数据的范围。然后,通过遍历每一行中的单元格,提取并打印出它们的数据。

根据条件筛选数据

除了按行或列提取数据外,我们还可以根据特定的条件来筛选数据。以下是一个根据条件筛选数据的示例:

1. from openpyxl import load_workbook  
2.   
3. # 加载Excel文件  
4. workbook = load_workbook('example.xlsx')  
5. sheet = workbook['Sheet1']  
6.   
7. # 根据条件筛选数据  
8. filtered_data = []  
9. for row in sheet.iter_rows(values_only=True):  # 提取所有行的数据,只包含值,不包含单元格对象  
10.     if row[0] > 100:  # 假设第一列是数值,筛选出大于100的行  
11.         filtered_data.append(row)  
12.   
13. # 打印筛选结果  
14. for data in filtered_data:  
15.     print(data)

在上面的代码中,我们首先使用iter_rows方法提取了所有行的数据,并设置values_only参数为True,以便只获取单元格的值而不包含单元格对象。然后,通过遍历每一行数据,并根据第一列的值进行条件判断,将符合条件的行添加到filtered_data列表中。最后,遍历filtered_data列表并打印出筛选结果。

四、生成新的Excel文件

在提取了指定数据后,我们可能需要将这些数据保存到一个新的Excel文件中。openpyxl库同样提供了创建和保存Excel文件的功能。以下是一个生成新Excel文件的示例:

1. from openpyxl import Workbook  
2.   
3. # 创建一个新的工作簿对象  
4. new_workbook = Workbook()  
5.   
6. # 获取活动工作表  
7. new_sheet = new_workbook.active  
8.   
9. # 假设我们已经从原始Excel文件中提取了数据,并将其保存在一个列表中  
10. # 列表中的每个元素也是一个列表,表示一行数据  
11. extracted_data = [  
12.     ['Name', 'Age', 'Gender'],  
13.     ['Alice', 25, 'Female'],  
14.     ['Bob', 30, 'Male'],  
15.     # ...更多数据
16. 
17. 
18. 将数据写入新的工作表
19. for row in extracted_data:
20.     new_sheet.append(row)
21. 
22.     保存新的Excel文件
23.     new_workbook.save('new_example.xlsx')

在上面的代码中,我们首先创建了一个新的工作簿对象new_workbook,并获取了其活动工作表new_sheet。然后,我们假设已经从原始Excel文件中提取了数据,并将其保存在一个名为extracted_data的列表中。接下来,我们使用append方法将每一行数据依次添加到新的工作表中。最后,通过调用save方法,将新的工作簿保存为一个名为new_example.xlsx的Excel文件。  

 

五、案例与完整代码示例  

 

下面是一个完整的案例,演示了如何使用openpyxl库从Excel文件中提取指定数据并生成新的Excel文件:  

 

假设我们有一个名为original_data.xlsx的Excel文件,其中包含员工的姓名、年龄和性别等信息。我们想要提取年龄大于30岁的员工信息,并将结果保存到一个新的Excel文件filtered_data.xlsx中。

 

首先,我们需要编写代码来提取年龄大于30岁的员工信息:  

 

1. from openpyxl import load_workbook  
2.   
3. # 加载原始Excel文件  
4. workbook = load_workbook('original_data.xlsx')  
5. sheet = workbook['Employee Data']  # 假设工作表名称为"Employee Data"  
6.   
7. # 提取年龄大于30岁的员工信息  
8. filtered_employees = []  
9. for row in sheet.iter_rows(min_row=2, values_only=True):  # 假设第一行是标题行,从第二行开始读取数据  
10.     age = row[1]  # 假设年龄位于第二列  
11.     if age > 30:  
12.         filtered_employees.append(row)  
13.   
14. # 打印提取结果(可选)  
15. for employee in filtered_employees:  
16.     print(employee)

接下来,我们将提取到的员工信息保存到一个新的Excel文件中:

1. from openpyxl import Workbook  
2.   
3. # 创建一个新的工作簿对象  
4. new_workbook = Workbook()  
5. new_sheet = new_workbook.active  
6.   
7. # 将提取到的员工信息写入新的工作表  
8. new_sheet.append(['Name', 'Age', 'Gender'])  # 写入标题行  
9. for employee in filtered_employees:  
10.     new_sheet.append(employee)  
11.   
12. # 保存新的Excel文件  
13. new_workbook.save('filtered_data.xlsx')

将以上两段代码结合起来,我们就可以实现从原始Excel文件中提取指定数据并生成新的Excel文件的功能。

六、总结

本文详细介绍了如何使用openpyxl库从Excel文件中提取指定的数据并生成新的文件。通过案例和代码示例,帮助新手朋友掌握了openpyxl库的基本用法和常见操作。在实际应用中,我们可以根据具体需求调整提取数据的条件和方式,以及生成新文件的格式和内容。

人生苦短,我用python

微信名片

 

显示推荐内容

目录
相关文章
|
1月前
|
Python
使用OpenPyXL库实现Excel单元格其他对齐方式设置
本文介绍了如何使用Python的`openpyxl`库设置Excel单元格中的文本对齐方式,包括文本旋转、换行、自动调整大小和缩进等,通过具体示例代码展示了每种对齐方式的应用方法,适合需要频繁操作Excel文件的用户学习参考。
175 85
使用OpenPyXL库实现Excel单元格其他对齐方式设置
|
30天前
|
人工智能 自然语言处理 Java
FastExcel:开源的 JAVA 解析 Excel 工具,集成 AI 通过自然语言处理 Excel 文件,完全兼容 EasyExcel
FastExcel 是一款基于 Java 的高性能 Excel 处理工具,专注于优化大规模数据处理,提供简洁易用的 API 和流式操作能力,支持从 EasyExcel 无缝迁移。
128 9
FastExcel:开源的 JAVA 解析 Excel 工具,集成 AI 通过自然语言处理 Excel 文件,完全兼容 EasyExcel
|
1月前
|
存储 Java easyexcel
招行面试:100万级别数据的Excel,如何秒级导入到数据库?
本文由40岁老架构师尼恩撰写,分享了应对招商银行Java后端面试绝命12题的经验。文章详细介绍了如何通过系统化准备,在面试中展示强大的技术实力。针对百万级数据的Excel导入难题,尼恩推荐使用阿里巴巴开源的EasyExcel框架,并结合高性能分片读取、Disruptor队列缓冲和高并发批量写入的架构方案,实现高效的数据处理。此外,文章还提供了完整的代码示例和配置说明,帮助读者快速掌握相关技能。建议读者参考《尼恩Java面试宝典PDF》进行系统化刷题,提升面试竞争力。关注公众号【技术自由圈】可获取更多技术资源和指导。
|
1月前
|
数据可视化 Python
使用OpenPyXL在Excel中创建折线图:数据可视化入门
本文介绍了如何使用Python的`openpyxl`库在Excel中创建折线图,包括安装库、加载Excel文件、定义数据范围、设置图表属性(如标题、轴标签)及保存文件等步骤,适合数据可视化初学者。
83 15
|
1月前
|
BI Python
利用OpenPyXL实现Excel条件格式化
本文介绍如何使用Python的`openpyxl`库为Excel文件添加条件格式,包括颜色渐变、图标集、数据条及基于公式的规则等,提升数据可读性和美观度。通过具体示例,展示了从安装库、加载文件到应用各种条件格式的详细过程,最后保存修改后的文件。
88 12
|
1月前
|
Python
批量将不同的工作簿合并到同一个Excel文件
本文介绍如何使用Python的`pandas`库批量合并不同工作簿至同一Excel文件。通过模拟生成三个班级的成绩数据,分别保存为Excel文件,再将这些文件合并成一个包含所有班级成绩的总成绩单。步骤包括安装必要库、生成数据、保存与合并工作簿。
61 6
|
1月前
|
Python
按条件将Excel文件拆分到不同的工作表
使用Python的pandas库,可以轻松将Excel文件按条件拆分为多个工作表。本文通过一个具体示例,展示了如何根据学生班级将成绩数据拆分到不同的工作表中,并生成一个包含总成绩表和各班级成绩表的Excel文件。代码简洁明了,适合初学者学习和应用。
54 6
|
1月前
|
数据采集 数据可视化 数据挖掘
利用Python自动化处理Excel数据:从基础到进阶####
本文旨在为读者提供一个全面的指南,通过Python编程语言实现Excel数据的自动化处理。无论你是初学者还是有经验的开发者,本文都将帮助你掌握Pandas和openpyxl这两个强大的库,从而提升数据处理的效率和准确性。我们将从环境设置开始,逐步深入到数据读取、清洗、分析和可视化等各个环节,最终实现一个实际的自动化项目案例。 ####
213 10
|
3月前
|
数据采集 存储 JavaScript
自动化数据处理:使用Selenium与Excel打造的数据爬取管道
本文介绍了一种使用Selenium和Excel结合代理IP技术从WIPO品牌数据库(branddb.wipo.int)自动化爬取专利信息的方法。通过Selenium模拟用户操作,处理JavaScript动态加载页面,利用代理IP避免IP封禁,确保数据爬取稳定性和隐私性。爬取的数据将存储在Excel中,便于后续分析。此外,文章还详细介绍了Selenium的基本设置、代理IP配置及使用技巧,并探讨了未来可能采用的更多防反爬策略,以提升爬虫效率和稳定性。
233 4
|
5月前
|
关系型数据库 MySQL Shell
不通过navicat工具怎么把查询数据导出到excel表中
不通过navicat工具怎么把查询数据导出到excel表中
65 0