python学习笔记--将excel中的数据转换为列表嵌套字典的格式

本文链接：https://blog.csdn.net/qingliuun/article/details/113877392

步骤一、获取列表表头信息；

步骤二、生成{表头:数据}的字典sub_data；

步骤三、将该字典添加到列表中。

from openpyxl import load_workbook


class DoExcel():

    def __init__(self, file_name, sheet_name):
        self.sheet_name = sheet_name
        self.file_name = file_name

    def get_header(self):
        """获取第一行的标题"""
        wb = load_workbook(self.file_name)
        sheet = wb[self.sheet_name]
        header = []  # 用来存储标题行
        for j in range(1, sheet.max_column + 1):
            header.append(sheet.cell(1, j).value)  # 添加表头到列表，把第一行的每一列title加到列表里
        return header

    def get_data(self):
        wb = load_workbook(self.file_name)
        sheet = wb[self.sheet_name]
        test_data = []  # 设置一个空列表，用来存储excel中所有行的数据
        header = self.get_header()  # 获取表头信息
        for i in range(2, sheet.max_row + 1):  # 排除表头，遍历每一行数据
            sub_data = {}  # 设置一个空字典，用来存储每一行和表头的对应关系
            for j in range(1, sheet.max_column + 1):  # 从第一列开始，遍历每一列的数据
                sub_data[header[j - 1]] = sheet.cell(i,
                                                     j).value  # 列表从0计数，j-1将每一列数据和表头header中的每个字段对应，生成一个{表头:数据}的字典sub_data
            test_data.append(sub_data)
        return test_data


if __name__ == '__main__':
    test_data = DoExcel(file_name, sheet_name).get_data()
    print(test_data)