Excel 转化成JSON

Excel 转化成JSON

import pandas as pd
import json
import os

def excel_to_json(excel_path, sheet_name=0, orient='records', save_path=None):
    """
    将Excel文件转换为JSON格式并可选择保存到文件
    
    参数:
        excel_path: Excel文件路径
        sheet_name: 工作表名称或索引，默认为第一个工作表
        orient: JSON格式取向（'records'、'split'、'index'等），默认为'records'
        save_path: JSON文件保存路径，如果为None则不保存文件
    
    返回:
        JSON字符串
    """
    # 读取Excel文件
    df = pd.read_excel(excel_path, sheet_name=sheet_name)
    
    # 处理NaN值，替换为None以确保正确转换为JSON的null
    df = df.where(df.notnull(), None)
    
    # 转换为JSON字符串
    json_str = df.to_json(orient=orient, force_ascii=False, indent=2)
    
    # 如果指定了保存路径，则保存到文件
    if save_path:
        # 确保目录存在
        os.makedirs(os.path.dirname(save_path), exist_ok=True)
        
        # 保存到文件
        with open(save_path, 'w', encoding='utf-8') as f:
            f.write(json_str)
        print(f"JSON文件已保存至: {save_path}")
    
    return json_str

# 使用示例
if __name__ == "__main__":
    # 转换Excel并保存为JSON文件
    json_output = excel_to_json(
        excel_path="data.xlsx",
        sheet_name="Sheet1",
        orient='records',
        save_path="output/data.json"  # 指定保存路径
    )
    
    # 打印部分JSON内容预览
    print("JSON内容预览:")
    print(json_output[:200] + "..." if len(json_output) > 200 else json_output)