Pandas 如何使用openpyxl模块将Pandas中的数据写入Excel文件

在本文中，我们将介绍如何使用openpyxl模块将Pandas中的数据写入Excel文件，并对所有单元格应用wrap_text格式。wrap_text格式允许文本自动换行，以适应单元格的宽度。

pandas.DataFrame.to_excel()和openpyxl

Pandas的to_excel()方法可用于将数据写入Excel文件。我们可以将DataFrame对象作为参数传递，然后将文件名称和工作表名称指定为选项。

import pandas as pd

# 创建示例DataFrame对象
data = {'Name': ['Tom', 'Jerry', 'Mickey'],
        'ID': [101, 102, 103],
        'Age': [25, 30, 35]}
df = pd.DataFrame(data)

# 写入Excel文件
writer = pd.ExcelWriter('output.xlsx', engine='openpyxl')
df.to_excel(writer, sheet_name='Sheet1')
writer.save()

在上面的代码中，我们创建了一个DataFrame对象，然后使用to_excel()方法将其写入名为“output.xlsx”的Excel文件的工作表“Sheet1”中。请注意，我们需要安装openpyxl模块才能使用此方法。

使用openpyxl为单元格应用wrap_text格式

我们可以在向Excel文件写入数据之前为DataFrame对象中的所有单元格应用wrap_text格式。要执行此操作，我们需要使用openpyxl模块打开Excel文件并获取工作表对象。然后，我们可以遍历所有单元格并将wrap_text格式应用于每个单元格。

from openpyxl import load_workbook
from openpyxl.styles import Alignment

# 打开Excel文件并获取工作表对象
wb = load_workbook(filename='output.xlsx')
ws = wb['Sheet1']

# 获取所有单元格
cells = ws.get_cell_collection()

# 应用wrap_text格式
wrap_alignment = Alignment(wrap_text=True)
for cell in cells:
    cell.alignment = wrap_alignment

# 保存工作表并关闭Excel文件
wb.save('output.xlsx')

在这里，我们使用load_workbook()方法打开名为“output.xlsx”的Excel文件，然后使用get_cell_collection()方法获取所有单元格。请注意，这将返回一个包含所有单元格对象的列表。我们创建了一个Alignment对象并将wrap_text参数设置为True，以启用wrap_text格式。然后，我们遍历所有单元格并将包含wrap_text格式的Alignment对象应用于每个单元格。