欢迎访问宙启技术站
智能推送

Python中使用writer()函数将数据写入Excel文件并设置格式

发布时间:2024-01-13 18:15:33

在Python中,可以使用openpyxl库来操作Excel文件。openpyxl是一个用于读取和写入Excel文件的库,它支持Excel文件的各种操作,包括写入数据、设置格式、合并单元格等。

要使用openpyxl库,首先需要安装该库。可以使用pip命令进行安装:

pip install openpyxl

安装完成后,就可以使用openpyxl库来操作Excel文件了。下面是一个使用writer()函数将数据写入Excel文件并设置格式的例子:

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

def write_data_to_excel(filename, data):
    # 创建一个新的Excel工作簿
    workbook = openpyxl.Workbook()
    # 获取工作簿的默认工作表
    sheet = workbook.active
    
    # 写入数据
    for row in data:
        sheet.append(row)
    
    # 设置单元格格式
    font = Font(name='Arial', bold=True)
    border = Border(left=Side(border_style='thin'),
                    right=Side(border_style='thin'),
                    top=Side(border_style='thin'),
                    bottom=Side(border_style='thin'))
    for row in sheet.iter_rows():
        for cell in row:
            cell.font = font
            cell.border = border
    
    # 保存Excel文件
    workbook.save(filename)
    print(f"数据已写入Excel文件'{filename}'")

# 测试数据
data = [
    ['Name', 'Age', 'Gender'],
    ['John', 25, 'Male'],
    ['Lisa', 30, 'Female'],
    ['Mike', 35, 'Male']
]

# 调用函数,将数据写入Excel文件并设置格式
write_data_to_excel('data.xlsx', data)

在上面的例子中,我们首先导入openpyxl库的Workbook类和Font、Border、Side类,它们用于操作Excel文件和设置单元格格式。

然后,我们定义了一个write_data_to_excel()函数,该函数接受一个文件名和一个数据列表作为参数。函数首先创建一个新的Excel工作簿,并获取其默认工作表。然后,它遍历数据列表,将每个元素写入工作表的新行。

接下来,我们设置了单元格的字体和边框样式。字体样式使用Font类来定义,可以设置字体名称、是否加粗等。边框样式使用Border和Side类来定义,可以设置边框样式、边框颜色等。在循环中,我们使用iter_rows()方法来遍历工作表的所有行,并使用font和border属性来设置单元格的样式。

最后,我们使用save()方法将工作簿保存为Excel文件,并打印提示信息。

运行以上代码后,将会生成一个名为'data.xlsx'的Excel文件,其中包含了我们定义的数据,并且每个单元格都设置了字体和边框样式。

这只是一个简单的例子,openpyxl库还提供了很多其他功能,例如合并单元格、设置列宽、设置单元格数值格式等。可以通过查阅openpyxl库的官方文档来了解更多详细信息。