如何使用Python处理Excel文件|极客笔记

如何使用Python处理Excel文件

在日常工作中，我们经常会遇到需要处理Excel文件的情况，比如读取数据、修改数据、保存数据等等。Python作为一种强大的编程语言，提供了很多库可以方便地处理Excel文件。本文将详细介绍如何使用Python处理Excel文件。

安装所需库

在开始之前，我们需要安装几个Python库，分别是openpyxl和pandas。可以通过以下命令来安装：

pip install openpyxl pandas

openpyxl是一个专门用来处理Excel文件的库，而pandas是一个数据分析库，在处理Excel数据的时候也非常方便。

读取Excel文件

首先，让我们来看看如何读取Excel文件。使用openpyxl库可以很方便地读取Excel文件中的数据。

from openpyxl import load_workbook

# 打开Excel文件
wb = load_workbook('example.xlsx')

# 获取当前活动的工作表
sheet = wb.active

# 读取数据
for row in sheet.iter_rows(values_only=True):
    print(row)

上面的代码首先打开了一个名为example.xlsx的Excel文件，然后获取了该Excel文件的活动工作表，并通过iter_rows方法循环遍历了每一行的数据并打印出来。

写入Excel文件

除了读取Excel文件外，我们经常需要将处理后的数据写入到Excel文件中。使用openpyxl库同样可以很方便地实现这一功能。

from openpyxl import Workbook

# 创建一个新的工作簿
wb = Workbook()

# 创建一个工作表
sheet = wb.active

# 写入数据
sheet['A1'] = 'Hello'
sheet['B1'] = 'World'

# 保存Excel文件
wb.save('output.xlsx')

上面的代码首先创建了一个新的Excel工作簿，然后获取了该工作簿的活动工作表，并将数据Hello和World写入到A1和B1单元格中，最后保存了该Excel文件为output.xlsx。

使用pandas处理Excel文件

除了openpyxl库外，pandas库也是一个非常方便处理Excel文件的工具。使用pandas库可以更加高效地处理Excel文件中的数据。

import pandas as pd

# 读取Excel文件
df = pd.read_excel('example.xlsx')

# 打印前5行数据
print(df.head())

# 写入Excel文件
df.to_excel('output.xlsx', index=False)

上面的代码首先使用pandas的read_excel方法读取了example.xlsx中的数据，并将其存储在一个DataFrame中。然后通过head方法打印了DataFrame的前5行数据，并最后通过to_excel方法将DataFrame中的数据写入到output.xlsx中。