美文网首页
Python-Excel表格数据转json

Python-Excel表格数据转json

作者: calary | 来源:发表于2022-05-16 15:28 被阅读0次

    一、前言

    开发中碰到产品给到的Excel表格数据,需要根据数据做对应的操作处理,就需要把数据转为json使用,为了方便高效的处理,百度了下,发现Python可以方便的处理这种转化问题。

    二、准备工作

    • 环境搭建:参考菜鸟, 或者自己百度安装
    • IDE: 推荐Pycharm,或者使用VS Code也可以,Pycharm更适合新手,可以直接创建项目,我们这里使用Pycharm

    三、代码编写

    • 打开Pycharm->New Project

    • 将test.xlsx和项目的main.py放在同一级


      项目结构.png
      excel.png
    • 安装三方库 openpyxl : pip install openpyxl

    • 具体代码,代码中有详细注释

    import openpyxl
    import json
    import io
    # excel表格转json文件
    def excel_to_json(excel_file, json_file_name):
        # 加载工作薄
        book = openpyxl.load_workbook(excel_file)
        # 获取sheet页
        sheet = book["Sheet1"]
        # 行数
        max_row = sheet.max_row
        # 列数
        max_column = sheet.max_column
        print("max_row: %d, max_column: %d" % (max_row, max_column))
        # 结果,数组存储
        result = []
        heads = []
        # 解析表头
        for column in range(max_column):
            # 读取的话行列是从(1,1)开始
            heads.append(sheet.cell(1, column + 1).value)
        # 遍历每一行
        for row in range(max_row):
            if row == 0:
                continue
            one_line = {}
            for column in range(max_column):
                # 读取第二行开始每一个数据
                k = heads[column]
                cell = sheet.cell(row + 1, column + 1)
                value = cell.value
                one_line[k] = value
            print(one_line)
            result.append(one_line)
        book.close()
        # 将json保存为文件
        save_json_file(result, json_file_name)
    
    # 将json保存为文件
    def save_json_file(jd, json_file_name):
        file = io.open(json_file_name, 'w', encoding='utf-8')
        # 把对象转化为json对象
        # indent: 参数根据数据格式缩进显示,读起来更加清晰
        # ensure_ascii = True:默认输出ASCII码,如果把这个该成False, 就可以输出中文。
        txt = json.dumps(jd, indent=2, ensure_ascii=False)
        file.write(txt)
        file.close()
    if '__main__' == __name__:
         excel_to_json(u'test.xlsx', 'result.json')
    
    
    

    运行后生成文件result.json,结果如下

    [
      {
        "name": "张三",
        "age": 22,
        "sex": "男"
      },
      {
        "name": "李四",
        "age": 30,
        "sex": "男"
      },
      {
        "name": "王阿五",
        "age": 18,
        "sex": "女"
      }
    ]
    

    四、进阶(有合并单元格的情况)

    目前只是简单的实现了最基础的数据结构,如果有合并单元格的情况出现,并不能很好的处理,合并单元格非左上角的其他单元格都会获取到None值,数据如图所示,B3,B4是合并的单元格,C4,C5是合并的单元格:

    excel.png
    运行后的结果为:
    max_row: 5, max_column: 3
    {'name': '张三', 'age': 22, 'sex': '男'}
    {'name': '李四', 'age': 30, 'sex': '男'}
    {'name': '王阿五', 'age': None, 'sex': '女'}
    {'name': '赵六', 'age': 12, 'sex': None}```
    
    • 解决思路

    获取到对应单元格后,判断该单元格是否为合并单元格,如果是,则找到该合并区域并获取左上角的值返回。

    import openpyxl
    import json
    import io
    from openpyxl.worksheet.worksheet import Worksheet
    from openpyxl.cell import MergedCell
    
    # excel表格转json文件
    def excel_to_json(excel_file, json_file_name):
        # 加载工作薄
        book = openpyxl.load_workbook(excel_file)
        # 获取sheet页
        sheet = book["Sheet1"]
        # 行数
        max_row = sheet.max_row
        # 列数
        max_column = sheet.max_column
        print("max_row: %d, max_column: %d" % (max_row, max_column))
        # 结果,数组存储
        result = []
        heads = []
        # 解析表头
        for column in range(max_column):
            # 读取的话行列是从(1,1)开始
            heads.append(sheet.cell(1, column + 1).value)
        # 遍历每一行
        for row in range(max_row):
            if row == 0:
                continue
            one_line = {}
            for column in range(max_column):
                # 读取第二行开始每一个数据
                k = heads[column]
                cell = parser_merged_cell(sheet, row + 1, column + 1)
                value = cell.value
                one_line[k] = value
            print(one_line)
            result.append(one_line)
        book.close()
        # 将json保存为文件
        save_json_file(result, json_file_name)
    
    # 将json保存为文件
    def save_json_file(jd, json_file_name):
        file = io.open(json_file_name, 'w', encoding='utf-8')
        # 把对象转化为json对象
        # indent: 参数根据数据格式缩进显示,读起来更加清晰
        # ensure_ascii = True:默认输出ASCII码,如果把这个该成False, 就可以输出中文。
        txt = json.dumps(jd, indent=2, ensure_ascii=False)
        file.write(txt)
        file.close()
    
    # 处理合并单元格
    def parser_merged_cell(sheet: Worksheet, row, column):
        cell = sheet.cell(row, column)
        # 判断该单元格是否为合并单元格
        if isinstance(cell, MergedCell):
            # 循环查找该单元格所属的合并区域
            for merged_range in sheet.merged_cells.ranges:
                if cell.coordinate in merged_range:
                     # 获取合并区域左上角的单元格作为该单元格的值返回
                    cell = sheet.cell(merged_range.min_row, merged_range.min_col)
                    break
        return cell
    
    if '__main__' == __name__:
         excel_to_json(u'test.xlsx', 'result.json')
    
    

    运行结果为

    max_row: 5, max_column: 3
    {'name': '张三', 'age': 22, 'sex': '男'}
    {'name': '李四', 'age': 30, 'sex': '男'}
    {'name': '王阿五', 'age': 30, 'sex': '女'}
    {'name': '赵六', 'age': 12, 'sex': '女'}
    

    五、其他情况处理

    • 二级复杂对象处理:在one_line处直接进行数据包裹等处理
    • 联表处理,可以根据sheet1,sheet2等先查询出来结果,然后再关联查询

    六、其他

    相关文章

      网友评论

          本文标题:Python-Excel表格数据转json

          本文链接:https://www.haomeiwen.com/subject/asgiurtx.html