代码收藏家技术教程 2024-09-26

Python导出数据生成Excel文件的方法与步骤

前言

在Python中，将数据导出生成Excel文件，最常用的库之一是openpyxl（用于处理.xlsx文件）和pandas（提供了更高级的数据处理功能，并且可以很方便地将DataFrame导出到Excel）。以下将分别介绍如何使用这两个库来生成Excel文件。

一、使用openpyxl生成Excel文件

如果你只想使用openpyxl来生成Excel文件，你可以手动创建工作簿、工作表，并写入数据。

from openpyxl import Workbook  
  
# 创建一个工作簿  
wb = Workbook()  
  
# 激活工作表  
ws = wb.active  
  
# 更改工作表的标题  
ws.title = "Example Sheet"  
  
# 写入数据  
ws.append([1, 2, 3])  # 写入一行数据  
ws['A2'] = 'Hello'  # 直接指定单元格写入数据  
ws.cell(row=3, column=2, value="World")  # 通过行号和列号写入数据  
  
# 保存工作簿  
wb.save("example.xlsx")

二、使用pandas生成Excel文件

如果你的数据已经是pandas DataFrame的形式，那么将数据导出到Excel会非常简单。

import pandas as pd  
  
# 创建一个DataFrame  
data = {  
    'Name': ['John', 'Anna', 'Peter', 'Linda'],  
    'Age': [28, 34, 29, 32],  
    'City': ['New York', 'Paris', 'Berlin', 'London']  
}  
df = pd.DataFrame(data)  
  
# 将DataFrame导出到Excel文件  
df.to_excel("output.xlsx", index=False)  # index=False表示不导出行索引

在这个例子中，我们首先创建了一个包含人名、年龄和城市的DataFrame。然后，我们使用to_excel方法将其导出到名为output.xlsx的Excel文件中，并通过设置index=False来避免导出DataFrame的行索引。

pandas的to_excel方法非常强大，它支持多种自定义选项，比如指定工作表名称、设置列宽、应用样式等。你可以查阅pandas的官方文档来了解更多高级功能。