py实现CSV转json

import sys
import json
import csv
import chardet

# input_file = sys.argv[1]
input_file =   "./tab_airspace_data.csv"
lines = ""

def detect_encoding(file_path):
    with open(file_path, 'rb') as f:
        result = chardet.detect(f.read())
    return result['encoding']
# 使用检测到的编码来读取文件
encoding = detect_encoding(input_file)

# 读取文件
with open(input_file, "r",encoding= encoding) as f:
    reader = csv.reader(f)
    # 创建一个空列表
    keys = []  
    datas = []
    index = 0
    # 遍历每一行,包含索引
    for row in reader:
        # 把数组row用#连接成字符串
        if(index == 0):
            keys = row
        else:
            
          datas.append(dict(zip(keys, row))) 
        index+=1  
# 序列化时对中文默认使用的ascii编码.想输出真正的中文需要指定ensure_ascii=False

json_str = json.dumps(datas, ensure_ascii=False, indent=4)
# 去除\",\\N,\n 无关符号
result_data = json_str.replace(r'\"','').replace(r'\\N','').replace(r'\n','')
output_file = input_file.replace("csv", "json")
# 写入文件
with open(output_file, "w", encoding="utf-8") as f:
    f.write(result_data)
    print("convert success")