Python创建xlsx|极客笔记

Python创建xlsx

1. 介绍

在数据处理和文档生成过程中，我们经常需要将数据以Excel的格式保存。Python中提供了多个库来处理Excel文件，其中最常用的是openpyxl库。

本文将详细介绍如何使用Python的openpyxl库来创建和操作Excel文件，并提供一些示例代码。

2. 安装

在开始之前，我们需要先安装openpyxl库。可以使用以下命令通过pip来安装：

pip install openpyxl

3. 创建工作簿和表格

首先，我们需要创建一个Excel工作簿，然后在工作簿中创建一个表格（也称为工作表）。可以使用以下代码创建工作簿和表格：

from openpyxl import Workbook

# 创建工作簿
workbook = Workbook()

# 获取默认活动表格
sheet = workbook.active

# 修改表格名称
sheet.title = "数据表"

# 保存工作簿
workbook.save("data.xlsx")

运行以上代码，将会在当前目录下创建名为data.xlsx的Excel文件，并在文件中创建一个名为”数据表”的表格。

4. 写入数据

接下来，我们可以向表格中写入数据。openpyxl库提供了多个方法来写入数据，包括按行、按列、按单元格等方式。

4.1 按行写入数据

from openpyxl import Workbook

workbook = Workbook()
sheet = workbook.active

# 写入数据
data = [
    ["Name", "Age", "Gender"],
    ["Tom", 25, "Male"],
    ["Lucy", 23, "Female"],
    ["John", 27, "Male"]
]

for row in data:
    sheet.append(row)

# 保存工作簿
workbook.save("data.xlsx")

以上代码将按行写入数据到表格中。运行后，打开data.xlsx文件，可以看到数据已经成功写入到工作簿中。

4.2 按列写入数据

from openpyxl import Workbook

workbook = Workbook()
sheet = workbook.active

# 写入数据
data = [
    ["Name", "Tom", "Lucy", "John"],
    ["Age", 25, 23, 27],
    ["Gender", "Male", "Female", "Male"]
]

for column in data:
    sheet.append(column)

# 保存工作簿
workbook.save("data.xlsx")

以上代码将按列写入数据到表格中。运行后，打开data.xlsx文件，可以看到数据已经成功写入到工作簿中。

4.3 按单元格写入数据

from openpyxl import Workbook

workbook = Workbook()
sheet = workbook.active

# 写入数据
sheet["A1"].value = "Name"
sheet["B1"].value = "Age"
sheet["C1"].value = "Gender"

sheet["A2"].value = "Tom"
sheet["B2"].value = 25
sheet["C2"].value = "Male"

sheet["A3"].value = "Lucy"
sheet["B3"].value = 23
sheet["C3"].value = "Female"

sheet["A4"].value = "John"
sheet["B4"].value = 27
sheet["C4"].value = "Male"

# 保存工作簿
workbook.save("data.xlsx")

以上代码将按单元格写入数据到表格中。运行后，打开data.xlsx文件，可以看到数据已经成功写入到工作簿中。

5. 格式设置

openpyxl库还提供了一些方法来设置单元格的格式，包括字体样式、背景颜色、边框等。

from openpyxl import Workbook
from openpyxl.styles import Font, Color, Border, Side

workbook = Workbook()
sheet = workbook.active

# 写入数据
data = [
    ["Name", "Age", "Gender"],
    ["Tom", 25, "Male"],
    ["Lucy", 23, "Female"],
    ["John", 27, "Male"]
]

for row in data:
    sheet.append(row)

# 设置表头样式
header_font = Font(bold=True)
header_fill = Color(rgb="FFC0CB")
header_border = Border(bottom=Side(border_style="thick"))
for cell in sheet[1]:
    cell.font = header_font
    cell.fill = header_fill
    cell.border = header_border

# 保存工作簿
workbook.save("data.xlsx")

以上代码将设置表头的字体样式为粗体，背景颜色为粉色，底部边框为粗线。运行后，打开data.xlsx文件，可以看到表头的样式已经生效。