Python保存至Excel

本文将详细介绍如何使用Python将数据保存至Excel文件。Python作为一种强大的编程语言，提供了许多库和功能，可以轻松地将数据导出到Excel，方便数据分析和共享。

一、安装所需库

在使用Python保存至Excel之前，首先需要安装所需的库。可以使用pip（或conda）命令安装pandas和openpyxl库：

pip install pandas openpyxl

二、使用pandas保存至Excel

使用pandas库可以快速、简单地保存数据至Excel。下面是一个简单的示例：

import pandas as pd

# 创建数据
data = {'姓名': ['张三', '李四', '王五'],
        '年龄': [25, 30, 35],
        '性别': ['男', '女', '男']}

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

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

上述代码中，我们首先创建了一个字典data，包含了姓名、年龄和性别的数据。然后使用pandas的DataFrame将数据转换成表格。最后使用to_excel方法将DataFrame保存至Excel文件（文件名为data.xlsx），其中的index=False表示不保存行索引。

三、设置Excel样式

在保存数据至Excel时，有时候需要对Excel的样式进行一些调整，例如设置单元格格式、添加表头、设定列宽等。可以使用openpyxl库来实现这些功能。下面是一个示例：

import pandas as pd
from openpyxl import Workbook
from openpyxl.styles import Font, Alignment

# 创建数据
data = {'姓名': ['张三', '李四', '王五'],
        '年龄': [25, 30, 35],
        '性别': ['男', '女', '男']}

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

# 创建Workbook和Worksheet对象
wb = Workbook()
ws = wb.active

# 添加表头
header = ['姓名', '年龄', '性别']
ws.append(header)

# 设置表头样式
font = Font(bold=True)
for cell in ws[1]:
    cell.font = font

# 添加数据
for row in df.values:
    ws.append(row)

# 设置列宽
ws.column_dimensions['A'].width = 10
ws.column_dimensions['B'].width = 5
ws.column_dimensions['C'].width = 5

# 保存至Excel
wb.save('data_with_style.xlsx')

上述代码中，我们使用openpyxl库创建了一个Workbook对象和一个Worksheet对象。然后，首先添加表头，设置表头样式为粗体。接下来循环遍历DataFrame的每一行数据，并将其添加到Worksheet中。最后，通过设置列宽调整Excel的显示效果。最后调用save方法保存Workbook至Excel文件（文件名为data_with_style.xlsx）。

四、保存多个Sheet至Excel

有时候需要将多个数据保存到同一个Excel文件的不同Sheet中。可以使用pandas提供的ExcelWriter来实现这个功能。下面是一个示例：

import pandas as pd

# 创建数据1
data1 = {'姓名': ['张三', '李四', '王五'],
         '年龄': [25, 30, 35],
         '性别': ['男', '女', '男']}
df1 = pd.DataFrame(data1)

# 创建数据2
data2 = {'学科': ['语文', '数学', '英语'],
         '分数': [90, 95, 85],
         '排名': [1, 2, 3]}
df2 = pd.DataFrame(data2)

# 创建ExcelWriter对象
writer = pd.ExcelWriter('data_multi_sheet.xlsx', engine='xlsxwriter')

# 将DataFrame写入不同Sheet
df1.to_excel(writer, sheet_name='个人信息', index=False)
df2.to_excel(writer, sheet_name='成绩单', index=False)

# 保存Excel文件
writer.save()

上述代码中，我们首先创建了两个DataFrame对象df1和df2，分别表示个人信息和成绩单的数据。然后创建了一个ExcelWriter对象，并指定engine为xlsxwriter，用于保存数据至Excel。接下去，使用to_excel方法将两个DataFrame分别写入不同的Sheet中，并指定Sheet名称。最后调用save方法保存Excel文件（文件名为data_multi_sheet.xlsx）。

五、结语

通过本文的介绍，我们了解了如何使用Python将数据保存至Excel文件。使用pandas和openpyxl库，可以简单地实现数据的导出和Excel样式的调整，方便数据分析和共享。希望本文能对你有所帮助。