Remove tax units per household column from display

Author: MaxGhenis

us/states/ca/aei_rebate/aei_rebate_analysis.ipynb

@@ -118,7 +118,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 4,
+   "execution_count": null,
    "metadata": {
     "execution": {
      "iopub.execute_input": "2025-09-15T18:56:58.431738Z",
@@ -127,104 +127,8 @@
      "shell.execute_reply": "2025-09-15T18:56:58.440461Z"
     }
    },
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "\n",
-      "=== REBATE BASE CALCULATIONS (2026) ===\n"
-     ]
-    },
-    {
-     "data": {
-      "text/html": [
-       "<div>\n",
-       "<style scoped>\n",
-       "    .dataframe tbody tr th:only-of-type {\n",
-       "        vertical-align: middle;\n",
-       "    }\n",
-       "\n",
-       "    .dataframe tbody tr th {\n",
-       "        vertical-align: top;\n",
-       "    }\n",
-       "\n",
-       "    .dataframe thead th {\n",
-       "        text-align: right;\n",
-       "    }\n",
-       "</style>\n",
-       "<table border=\"1\" class=\"dataframe\">\n",
-       "  <thead>\n",
-       "    <tr style=\"text-align: right;\">\n",
-       "      <th></th>\n",
-       "      <th>Unit Type</th>\n",
-       "      <th>Total CA Units</th>\n",
-       "      <th>Units with Rebate</th>\n",
-       "      <th>Average Rebate Base</th>\n",
-       "      <th>Total Rebate Base</th>\n",
-       "      <th>Tax Units per Household</th>\n",
-       "    </tr>\n",
-       "  </thead>\n",
-       "  <tbody>\n",
-       "    <tr>\n",
-       "      <th>0</th>\n",
-       "      <td>household</td>\n",
-       "      <td>14.6M</td>\n",
-       "      <td>3.2M (22%)</td>\n",
-       "      <td>$20,245</td>\n",
-       "      <td>$65.3B</td>\n",
-       "      <td>1.50</td>\n",
-       "    </tr>\n",
-       "    <tr>\n",
-       "      <th>1</th>\n",
-       "      <td>tax_unit</td>\n",
-       "      <td>21.9M</td>\n",
-       "      <td>8.7M (40%)</td>\n",
-       "      <td>$18,146</td>\n",
-       "      <td>$158.4B</td>\n",
-       "      <td>1.50</td>\n",
-       "    </tr>\n",
-       "  </tbody>\n",
-       "</table>\n",
-       "</div>"
-      ],
-      "text/plain": [
-       "   Unit Type Total CA Units Units with Rebate Average Rebate Base  \\\n",
-       "0  household          14.6M        3.2M (22%)             $20,245   \n",
-       "1   tax_unit          21.9M        8.7M (40%)             $18,146   \n",
-       "\n",
-       "  Total Rebate Base Tax Units per Household  \n",
-       "0            $65.3B                    1.50  \n",
-       "1           $158.4B                    1.50  "
-      ]
-     },
-     "execution_count": 4,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "# Stack results into a single DataFrame for easy comparison\n",
-    "results_df = pd.DataFrame([household_results, tax_unit_results])\n",
-    "\n",
-    "# Format numeric columns for display\n",
-    "display_df = results_df.copy()\n",
-    "display_df['Total CA Units'] = display_df['total_ca_units'].apply(lambda x: f\"{x/1e6:.1f}M\")\n",
-    "display_df['Units with Rebate'] = display_df.apply(lambda r: f\"{r['units_with_rebate']/1e6:.1f}M ({r['rebate_percentage']:.0%})\", axis=1)\n",
-    "display_df['Average Rebate Base'] = display_df['average_rebate_base'].apply(lambda x: f\"${x:,.0f}\")\n",
-    "display_df['Total Rebate Base'] = display_df['total_rebate_base'].apply(lambda x: f\"${x/1e9:.1f}B\")\n",
-    "\n",
-    "# Add a simple ratio column\n",
-    "display_df['Ratio'] = f\"{tax_unit_results['total_ca_units']/household_results['total_ca_units']:.2f}\"\n",
-    "\n",
-    "# Select columns for display\n",
-    "display_cols = ['unit_type', 'Total CA Units', 'Units with Rebate', 'Average Rebate Base', 'Total Rebate Base', 'Ratio']\n",
-    "display_df = display_df[display_cols]\n",
-    "display_df.columns = ['Unit Type', 'Total CA Units', 'Units with Rebate', 'Average Rebate Base', 'Total Rebate Base', 'Tax Units per Household']\n",
-    "\n",
-    "print(f\"\\n=== REBATE BASE CALCULATIONS ({SIMULATION_YEAR}) ===\")\n",
-    "display_df"
-   ]
+   "outputs": [],
+   "source": "# Stack results into a single DataFrame for easy comparison\nresults_df = pd.DataFrame([household_results, tax_unit_results])\n\n# Format numeric columns for display\ndisplay_df = results_df.copy()\ndisplay_df['Total CA Units'] = display_df['total_ca_units'].apply(lambda x: f\"{x/1e6:.1f}M\")\ndisplay_df['Units with Rebate'] = display_df.apply(lambda r: f\"{r['units_with_rebate']/1e6:.1f}M ({r['rebate_percentage']:.0%})\", axis=1)\ndisplay_df['Average Rebate Base'] = display_df['average_rebate_base'].apply(lambda x: f\"${x:,.0f}\")\ndisplay_df['Total Rebate Base'] = display_df['total_rebate_base'].apply(lambda x: f\"${x/1e9:.1f}B\")\n\n# Select columns for display\ndisplay_cols = ['unit_type', 'Total CA Units', 'Units with Rebate', 'Average Rebate Base', 'Total Rebate Base']\ndisplay_df = display_df[display_cols]\ndisplay_df.columns = ['Unit Type', 'Total CA Units', 'Units with Rebate', 'Average Rebate Base', 'Total Rebate Base']\n\nprint(f\"\\n=== REBATE BASE CALCULATIONS ({SIMULATION_YEAR}) ===\")\ndisplay_df"
   },
   {
    "cell_type": "code",