Files
stockdb/new_write.py
2024-08-30 19:54:37 -04:00

46 lines
1.4 KiB
Python

import json
import pandas as pd
# Step 1: Load the JSON data
with open('CIK0000320193.json') as f:
data = json.load(f)
# Step 2: Extract the relevant fields
cik = data.get('cik')
entity_name = data.get('entityName')
facts = data.get('facts')
# Prepare a list to hold the rows
rows = []
# Traverse through the facts dictionary
for taxonomy, fact_items in facts.items():
for fact_id, fact_data in fact_items.items():
label = fact_data.get('label')
description = fact_data.get('description')
units = fact_data.get('units')
# For each unit and its details, add a new row
for unit, details_list in units.items():
for details in details_list:
row = {
'entity_cik': cik,
'entity_name': entity_name,
'fact_id': fact_id,
'fact_taxonomy': taxonomy,
'fact_label': label,
'fact_description': description,
'fact_unit': unit
}
# Include the additional details in the row
row.update(details)
# Append the row to the rows list
rows.append(row)
# Step 3: Create the DataFrame
df = pd.DataFrame(rows)
# Step 4: Output the head of the DataFrame
print(df.head())