python中数据如何存入excel

Python中将数据存入Excel的核心方法包括：使用pandas、openpyxl、xlwt模块。这些方法可以处理不同类型和规模的数据，并提供灵活的选项来格式化和操作Excel文件。以pandas为例，它不仅简化了数据处理流程，还能与Excel进行无缝集成。

下面将详细介绍如何使用这些模块将数据存入Excel文件。

一、使用pandas

1、安装pandas

首先需要安装pandas模块，可以使用以下命令：

pip install pandas

2、创建和保存DataFrame

pandas的DataFrame是一个二维表格结构，可以非常方便地将数据存储到Excel文件中。

import pandas as pd
创建一个示例DataFrame
data = {
    'Name': ['Alice', 'Bob', 'Charlie'],
    'Age': [25, 30, 35],
    'City': ['New York', 'San Francisco', 'Los Angeles']
}
df = pd.DataFrame(data)
将DataFrame保存到Excel文件
df.to_excel('output.xlsx', index=False)

在上面的代码中，to_excel方法将DataFrame保存到Excel文件中。index=False参数用于在保存时不包含行索引。

3、设置工作表和格式

可以指定保存到Excel文件中的工作表名称，并设置单元格的格式。

# 保存到不同的工作表
with pd.ExcelWriter('output.xlsx') as writer:
    df.to_excel(writer, sheet_name='Sheet1', index=False)
设置单元格格式
df.style.set_properties({'text-align': 'left'}).to_excel('output.xlsx', index=False)

二、使用openpyxl

1、安装openpyxl

安装openpyxl模块：

pip install openpyxl

2、创建和保存Excel文件

openpyxl模块可以直接操作Excel文件，适用于需要更多定制化操作的场景。

from openpyxl import Workbook
创建一个新的工作簿
wb = Workbook()
ws = wb.active
添加数据
ws['A1'] = 'Name'
ws['B1'] = 'Age'
ws['C1'] = 'City'
data = [
    ('Alice', 25, 'New York'),
    ('Bob', 30, 'San Francisco'),
    ('Charlie', 35, 'Los Angeles')
]
for row in data:
    ws.append(row)
保存工作簿
wb.save('output.xlsx')

在上面的代码中，首先创建一个新的工作簿，然后向工作表中添加数据，最后保存工作簿。

3、设置单元格样式

openpyxl还支持设置单元格的样式，如字体、颜色和对齐方式。

from openpyxl.styles import Font, Alignment
设置字体和对齐方式
font = Font(name='Arial', size=12, bold=True)
alignment = Alignment(horizontal='center', vertical='center')
应用样式
for cell in ws['1:1']:
    cell.font = font
    cell.alignment = alignment
保存工作簿
wb.save('output.xlsx')

三、使用xlwt

1、安装xlwt

安装xlwt模块：

pip install xlwt

2、创建和保存Excel文件

xlwt模块主要用于创建Excel 97-2003格式的文件（.xls）。

import xlwt
创建一个新的工作簿
wb = xlwt.Workbook()
ws = wb.add_sheet('Sheet1')
添加数据
ws.write(0, 0, 'Name')
ws.write(0, 1, 'Age')
ws.write(0, 2, 'City')
data = [
    ('Alice', 25, 'New York'),
    ('Bob', 30, 'San Francisco'),
    ('Charlie', 35, 'Los Angeles')
]
for i, row in enumerate(data, start=1):
    for j, value in enumerate(row):
        ws.write(i, j, value)
保存工作簿
wb.save('output.xls')

在上面的代码中，创建一个新的工作簿和工作表，然后向工作表中添加数据，最后保存工作簿。

3、设置单元格样式

可以设置单元格的字体、颜色和对齐方式。

# 设置字体和对齐方式
font = xlwt.Font()
font.name = 'Arial'
font.bold = True
alignment = xlwt.Alignment()
alignment.horz = xlwt.Alignment.HORZ_CENTER
alignment.vert = xlwt.Alignment.VERT_CENTER
style = xlwt.XFStyle()
style.font = font
style.alignment = alignment
应用样式
ws.write(0, 0, 'Name', style)
ws.write(0, 1, 'Age', style)
ws.write(0, 2, 'City', style)
保存工作簿
wb.save('output.xls')

四、综合实例

下面是一个综合实例，演示如何使用pandas、openpyxl和xlwt将数据存入Excel文件，并设置不同的格式和样式。

1、使用pandas

import pandas as pd
创建一个示例DataFrame
data = {
    'Name': ['Alice', 'Bob', 'Charlie'],
    'Age': [25, 30, 35],
    'City': ['New York', 'San Francisco', 'Los Angeles']
}
df = pd.DataFrame(data)
保存到Excel文件，并设置工作表名称和单元格格式
with pd.ExcelWriter('output.xlsx', engine='openpyxl') as writer:
    df.to_excel(writer, sheet_name='Sheet1', index=False)
    workbook = writer.book
    worksheet = writer.sheets['Sheet1']
    # 设置单元格格式
    for col in worksheet.columns:
        max_length = 0
        column = col[0].column_letter # 获取列字母
        for cell in col:
            try:
                if len(str(cell.value)) > max_length:
                    max_length = len(cell.value)
            except:
                pass
        adjusted_width = (max_length + 2)
        worksheet.column_dimensions[column].width = adjusted_width

2、使用openpyxl

from openpyxl import Workbook
from openpyxl.styles import Font, Alignment
创建一个新的工作簿
wb = Workbook()
ws = wb.active
添加数据
ws['A1'] = 'Name'
ws['B1'] = 'Age'
ws['C1'] = 'City'
data = [
    ('Alice', 25, 'New York'),
    ('Bob', 30, 'San Francisco'),
    ('Charlie', 35, 'Los Angeles')
]
for row in data:
    ws.append(row)
设置字体和对齐方式
font = Font(name='Arial', size=12, bold=True)
alignment = Alignment(horizontal='center', vertical='center')
应用样式
for cell in ws['1:1']:
    cell.font = font
    cell.alignment = alignment
保存工作簿
wb.save('output.xlsx')

3、使用xlwt

import xlwt
创建一个新的工作簿
wb = xlwt.Workbook()
ws = wb.add_sheet('Sheet1')
添加数据
ws.write(0, 0, 'Name')
ws.write(0, 1, 'Age')
ws.write(0, 2, 'City')
data = [
    ('Alice', 25, 'New York'),
    ('Bob', 30, 'San Francisco'),
    ('Charlie', 35, 'Los Angeles')
]
for i, row in enumerate(data, start=1):
    for j, value in enumerate(row):
        ws.write(i, j, value)
设置字体和对齐方式
font = xlwt.Font()
font.name = 'Arial'
font.bold = True
alignment = xlwt.Alignment()
alignment.horz = xlwt.Alignment.HORZ_CENTER
alignment.vert = xlwt.Alignment.VERT_CENTER
style = xlwt.XFStyle()
style.font = font
style.alignment = alignment
应用样式
ws.write(0, 0, 'Name', style)
ws.write(0, 1, 'Age', style)
ws.write(0, 2, 'City', style)
保存工作簿
wb.save('output.xls')