Python保存数据到Excel|极客教程

Python保存数据到Excel

在日常工作和数据处理中，我们经常需要将数据保存到Excel文件中。Python作为一种强大的编程语言，提供了许多库和工具来方便地处理Excel文件。在本文中，我们将介绍如何使用Python保存数据到Excel文件中。

使用openpyxl库保存数据到Excel

openpyxl是一个强大的Python库，用于读写Excel文件。我们可以使用它来创建一个新的Excel文件，并将数据保存到其中。首先，我们需要安装openpyxl库。可以使用以下命令来安装该库：

pip install openpyxl

接下来，我们可以使用以下代码示例来保存数据到Excel文件中：

from openpyxl import Workbook

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

# 选择默认的活动工作表
ws = wb.active

# 数据
data = [
    ['Name', 'Age'],
    ['Alice', 25],
    ['Bob', 30],
    ['Charlie', 35]
]

# 写入数据
for row in data:
    ws.append(row)

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

在上面的示例中，我们首先导入Workbook类，然后创建一个新的Excel文件对象。我们选择默认的活动工作表，并定义要保存的数据。接着，我们遍历数据并将其逐行写入工作表中。最后，我们使用save方法保存Excel文件。

运行上述代码后，将在当前目录下生成一个名为data.xlsx的Excel文件，并包含我们定义的数据。

使用pandas库保存数据到Excel

另一个常用的Python库是pandas，它提供了许多数据操作和分析工具。pandas库也可以用来保存数据到Excel文件中。首先需要安装pandas库：

pip install pandas

以下示例展示了如何使用pandas保存数据到Excel文件中：

import pandas as pd

# 数据
data = {
    'Name': ['Alice', 'Bob', 'Charlie'],
    'Age': [25, 30, 35]
}

# 创建DataFrame对象
df = pd.DataFrame(data)

# 保存数据到Excel
df.to_excel('data.xlsx', index=False)

在上面的示例中，我们首先定义了一个数据字典，然后使用pd.DataFrame函数创建一个DataFrame对象。最后，我们使用to_excel方法将数据保存到Excel文件中，并设置index=False以避免保存行索引。

运行上述代码后，将在当前目录下生成一个名为data.xlsx的Excel文件，并包含我们定义的数据。