Update Unnestjson.py

csuhaas · web-flow · commit 7fd43a8b6f62 · 2025-05-14T17:47:08.000+05:30
diff --git a/Unnestjson.py b/Unnestjson.py
@@ -1,23 +1,33 @@
 import json
 import pandas as pd
-from pandas import json_normalize
 
-# Load JSON from file or string
+# Recursive function to flatten nested JSON
+def flatten(obj, parent_key='', sep='.'):
+    items = []
+    if isinstance(obj, dict):
+        for k, v in obj.items():
+            new_key = f"{parent_key}{sep}{k}" if parent_key else k
+            items.extend(flatten(v, new_key, sep=sep).items())
+    elif isinstance(obj, list):
+        for i, v in enumerate(obj):
+            new_key = f"{parent_key}[{i}]"
+            items.extend(flatten(v, new_key, sep=sep).items())
+    else:
+        items.append((parent_key, obj))
+    return dict(items)
+
+# Load JSON file
 with open("input.json") as f:
-    data = json.load(f)
+    raw_data = json.load(f)
 
-# Flatten JSON
-def flatten_json(json_data):
-    if isinstance(json_data, list):
-        df = json_normalize(json_data)
-    elif isinstance(json_data, dict):
-        df = json_normalize([json_data])
-    else:
-        raise ValueError("Unsupported JSON format")
-    return df
+# Handle list of records or single record
+if isinstance(raw_data, list):
+    flat_data = [flatten(entry) for entry in raw_data]
+else:
+    flat_data = [flatten(raw_data)]
 
-# Unnest and convert to CSV
-df = flatten_json(data)
+# Convert to DataFrame
+df = pd.DataFrame(flat_data)
 
 # Save to CSV
 df.to_csv("output.csv", index=False)
