Python 处理 Excel 文件详解

三、写入 Excel 文件

除了读取 Excel 文件，我们还可以使用 Python 将数据写入 Excel 文件。以下是一个示例代码：

# 创建数据
data = {
    'Name': ['Alice', 'Bob', 'Charlie'],
    'Age': [25, 30, 35],
    'City': ['New York', 'London', 'Paris']
}

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

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

这段代码首先创建了一个包含姓名、年龄和城市的数据字典。然后，通过 pd.DataFrame() 创建 DataFrame 对象 df。最后，使用 to_excel() 方法将 DataFrame 写入到 output.xlsx 文件中。index=False 表示不将索引写入文件，sheet_name 指定工作表名称。

四、修改 Excel 文件

除了读取和写入，我们还可以使用 Python 修改 Excel 文件中的数据、添加新的 Sheet 等。以下是一个示例代码：

# 打开 Excel 文件
wb = openpyxl.load_workbook("data.xlsx")

# 获取指定 Sheet
sheet = wb["Sheet1"]

# 修改单元格数据
sheet["A1"] = "Updated Value"

# 添加新的 Sheet
new_sheet = wb.create_sheet("Sheet2")

# 保存修改后的 Excel 文件
wb.save("data_modified.xlsx")

这段代码首先使用 openpyxl.load_workbook() 方法打开 data.xlsx 文件，返回一个 Workbook 对象 wb。然后，通过指定 Sheet 的名称或索引获取指定的 Sheet。接下来，我们可以修改 Sheet 中的单元格数据，例如将 "A1" 单元格的值修改为 "Updated Value"。然后，使用 create_sheet() 方法添加一个新的 Sheet。最后，使用 wb.save() 方法保存修改后的 Excel 文件。

4.1 设置单元格样式

使用 openpyxl 还可以设置字体、颜色和边框：

from openpyxl.styles import Font, PatternFill, Border, Side

# 定义样式
font = Font(bold=True, color="FF0000")
fill = PatternFill(start_color="FFFF00", fill_type="solid")
border = Border(left=Side(style='thin'), right=Side(style='thin'))

# 应用样式
sheet['A1'].font = font
sheet['A1'].fill = fill
sheet['A1'].border = border

五、高级技巧与异常处理

5.1 大数据量处理

当处理大型 Excel 文件时，一次性加载可能导致内存溢出。建议使用迭代器模式逐行处理：

for row in wb.active.iter_rows(values_only=True):
    # 处理每一行数据
    print(row)

5.2 异常捕获

在处理文件时，应添加错误处理机制以防止程序崩溃：

try:
    data = pd.read_excel("data.xlsx")
except FileNotFoundError:
    print("文件未找到，请检查路径")
except Exception as e:
    print(f"发生错误：{e}")

六、总结

通过 pandas 和 openpyxl 等第三方库，我们可以方便地读取、写入和修改 Excel 文件。本文介绍了 Python 对 Excel 文件处理的基本方法，并提供了相应的代码示例和详细说明。你可以根据实际需求，进一步探索这些库的其他功能和特性。Excel 文件的处理能力将为你的 Python 应用程序带来更多可能性，帮助你更好地处理和分析数据。