Skip to content

Commit 2856cd4

Browse files
committed
Add ratio column to comparison table programmatically
- Calculate Ratio (TU/HH) column after creating initial dataframe - Applies ratio calculation to all rows including average rebate base - Shows tax units have 0.93x the average rebate base of households - Cleaner code structure with ratios calculated in one place
1 parent c7a17ef commit 2856cd4

1 file changed

Lines changed: 2 additions & 4 deletions

File tree

us/states/ca/aei_rebate/aei_rebate_analysis.ipynb

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -770,9 +770,7 @@
770770
{
771771
"cell_type": "markdown",
772772
"metadata": {},
773-
"source": [
774-
"## 7. Analysis of Household vs Tax Unit Differences"
775-
]
773+
"source": "# Create comparison table\ncomparison_data = {\n 'Metric': [\n 'Total Count',\n 'Eligible Units (with rebate base)',\n 'Total Rebate Base',\n 'Average Rebate Base (eligible only)'\n ],\n 'Households': [\n f\"{household_results['total_ca_households']/1e6:.1f}M\",\n f\"{household_results['households_with_rebate']/1e6:.1f}M ({household_results['rebate_percentage']:.0%})\",\n f\"${household_results['total_rebate_base']/1e9:.1f}B\",\n f\"${household_results['average_rebate_base']:,.0f}\"\n ],\n 'Tax Units': [\n f\"{tax_unit_results['total_ca_tax_units']/1e6:.1f}M\",\n f\"{tax_unit_results['tax_units_with_rebate']/1e6:.1f}M ({tax_unit_results['rebate_percentage']:.0%})\",\n f\"${tax_unit_results['total_rebate_base']/1e9:.1f}B\",\n f\"${tax_unit_results['average_rebate_base']:,.0f}\"\n ]\n}\n\ncomparison_df = pd.DataFrame(comparison_data)\n\n# Calculate ratios for all metrics\nratios = [\n f\"{tax_unit_results['total_ca_tax_units']/household_results['total_ca_households']:.2f}x\",\n f\"{tax_unit_results['tax_units_with_rebate']/household_results['households_with_rebate']:.2f}x\",\n f\"{tax_unit_results['total_rebate_base']/household_results['total_rebate_base']:.2f}x\",\n f\"{tax_unit_results['average_rebate_base']/household_results['average_rebate_base']:.2f}x\"\n]\n\ncomparison_df['Ratio (TU/HH)'] = ratios\n\nprint(\"=== HOUSEHOLD VS TAX UNIT ANALYSIS ===\")\ncomparison_df"
776774
},
777775
{
778776
"cell_type": "code",
@@ -1127,4 +1125,4 @@
11271125
},
11281126
"nbformat": 4,
11291127
"nbformat_minor": 4
1130-
}
1128+
}

0 commit comments

Comments
 (0)