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库的官方文档来了解更多详细信息。
