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

使用Python的write()函数将数据写入Excel文件的方法

发布时间:2023-12-14 08:13:33

Python中可以使用第三方库xlwt来操作Excel文件,其中write()函数可以将数据写入Excel文件中。

使用write()函数写入Excel文件的步骤如下:

1. 导入所需的xlwt库:import xlwt

2. 创建一个Workbook对象:workbook = xlwt.Workbook()

3. 创建一个Worksheet对象:worksheet = workbook.add_sheet('Sheet1')

4. 使用write()函数来写入数据:worksheet.write(row, col, data)

- row: 行索引,从0开始

- col: 列索引,从0开始

- data: 要写入的数据

5. 保存并关闭Excel文件:workbook.save('output.xls')

下面是一个完整的使用write()函数将数据写入Excel文件的示例代码:

import xlwt

def write_to_excel(data):
    workbook = xlwt.Workbook()
    worksheet = workbook.add_sheet('Sheet1')

    # 写入表头
    headers = ['姓名', '年龄', '性别']
    for col, header in enumerate(headers):
        worksheet.write(0, col, header)

    # 写入数据
    for row, row_data in enumerate(data):
        for col, cell_data in enumerate(row_data):
            worksheet.write(row+1, col, cell_data)

    # 保存Excel文件
    workbook.save('output.xls')

# 待写入的数据
data = [
    ['张三', 25, '男'],
    ['李四', 30, '女'],
    ['王五', 28, '男']
]

write_to_excel(data)

上述代码中,首先导入了xlwt库。然后定义了一个write_to_excel函数,该函数接受一个二维数据作为参数。

在函数内部,首先创建了一个Workbook对象和一个Worksheet对象。然后使用write()函数来写入表头,在最外层的for循环中,使用write()函数来逐行逐列写入数据。

最后保存Excel文件并关闭。运行上述代码后,将会生成一个名为"output.xls"的Excel文件,其中包含了待写入的数据。