换行文本xlsxwriter列pd dataframe python

import pandas as pd
import xlsxwriter

# 创建一个示例 DataFrame
data = {
    'Name': ['Alice', 'Bob', 'Charlie'],
    'Description': ['This is a long text\nthat needs to be wrapped.', 'Short text', 'Another long text\nwith multiple lines.']
}
df = pd.DataFrame(data)

# 创建一个 Excel 文件
writer = pd.ExcelWriter('output.xlsx', engine='xlsxwriter')

# 将 DataFrame 写入 Excel 文件
df.to_excel(writer, sheet_name='Sheet1', index=False)

# 获取 xlsxwriter 的 workbook 和 worksheet 对象
workbook = writer.book
worksheet = writer.sheets['Sheet1']

# 设置列宽
worksheet.set_column('A:A', 20)
worksheet.set_column('B:B', 40)

# 设置换行文本格式
format_wrap = workbook.add_format({'text_wrap': True})

# 应用换行文本格式
worksheet.set_row(0, None, format_wrap)
worksheet.set_row(1, None, format_wrap)
worksheet.set_row(2, None, format_wrap)

# 保存 Excel 文件
writer.save()

# 设置换行文本格式
format_wrap = workbook.add_format({'text_wrap': True})

# 应用换行文本格式
worksheet.set_row(0, None, format_wrap)
worksheet.set_row(1, None, format_wrap)
worksheet.set_row(2, None, format_wrap)