【Python从入门到实践5.0】将数据写入Excel表格

将数据写入Excel表格

功能：
将模拟数据按规则写入excel表格中

编程思路
1、检查xlsx文件是否存在，若不存在，则创建文件并保存。
2、打开xlsx文件，读取Sheet表。
3、定义表头（学号、姓名、年龄）
4、获取Sheet最大行+1。根据行、列循环写入模拟数据student，且每次循环行数+1
5、数据录入完成后保存

安装openpyxl第三方库pip install openpyxl

import os
from openpyxl import load_workbook
from openpyxl import Workbook

student={'19321':['小明',16],'19457':['小红',17],'19352':['小军',16]}

def xlsx_operation(data:dict):
    addr = "openpyxl.xlsx"
    if not os.path.exists(addr):
        # 实例化 
        wb = Workbook()  # 如果文件不存在，则创建文件
        # 保存文件
        wb.save(addr)
    wb = load_workbook(addr)  # 打开加载文件

    # 读取表
    ws = wb['Sheet']

    # 定义表头
    ws['A1'] = '学号'
    ws['B1'] = '姓名'
    ws['C1'] = '年龄'

    row_max = ws.max_row  # 获取当前Sheet表格的最大行
    for i,k in data.items():
        row_max += 1
        # 写入学号
        ws.cell(row=row_max, column=1).value = i
        # 写入姓名
        ws.cell(row=row_max, column=2).value = k[0]
        # 写入年龄
        ws.cell(row=row_max, column=3).value = k[1]

    wb.save(addr)
    print('录入完成')

xlsx_operation(data=student)