Open Bell-Fintech opened 1 year ago
import os import csv import json
csv_dir = "/path/to/csv/directory" csv_name = "example"
csv_path = os.path.join(csv_dir, f"{csv_name}.csv") with open(csv_path, "r") as f: reader = csv.reader(f) header = next(reader) data_type = [type(item).name for item in next(reader)] num_rows = sum(1 for row in reader) num_cols = len(header)
info = { "setname": csv_dir, "key": csv_name, "type": data_type, "shape": (num_rows, num_cols), "description": "" }
info_path = os.path.join(csv_dir, f"{csv_name}.info") with open(info_path, "w") as f: json.dump(info, f)
import os import csv import json
设置csv文件所在目录和文件名
csv_dir = "/path/to/csv/directory" csv_name = "example"
读取csv文件,获取数据类型和数据形状
csv_path = os.path.join(csv_dir, f"{csv_name}.csv") with open(csv_path, "r") as f: reader = csv.reader(f) header = next(reader) data_type = [type(item).name for item in next(reader)] num_rows = sum(1 for row in reader) num_cols = len(header)
生成csv.info文件内容
info = { "setname": csv_dir, "key": csv_name, "type": data_type, "shape": (num_rows, num_cols), "description": "" }
将csv.info内容保存为json文件
info_path = os.path.join(csv_dir, f"{csv_name}.info") with open(info_path, "w") as f: json.dump(info, f)