Script for converting output to csv

This commit is contained in:
Gusto 2023-08-15 02:36:22 +03:00
parent f6ae008e40
commit b8652c67a9
1 changed files with 28 additions and 0 deletions

28
scripts/data_to_csv.py Normal file
View File

@ -0,0 +1,28 @@
import pandas as pd
import json
import argparse
def data_to_csv(data_path, output_path):
header_written = False
with open(data_path, 'r') as f:
for line in f:
try:
clean_line = line.rstrip(",\n")
data = json.loads(clean_line)
normalized = pd.json_normalize(data)
normalized.to_csv(output_path, mode='a', header=not header_written, index=False)
# Set the header_written flag to True after the first write
header_written = True
except json.JSONDecodeError:
print(f"Failed to parse line: {line}")
if __name__ == "__main__":
parser = argparse.ArgumentParser(description="Normalize JSON lines in a file to a Pandas DataFrame and append to CSV.")
parser.add_argument("data_path", type=str, help="Path to the file containing JSON lines.")
parser.add_argument("output_path", type=str, help="Path to the output CSV file.")
args = parser.parse_args()
data_to_csv(args.data_path, args.output_path)