Cosmetic changes to finalised demo_jt notebook

notebooks/user/vbui/demo_jt.ipynb

@@ -51,6 +51,7 @@
     "    \"\"\"\n",
     "    ages = range(upper_age)\n",
     "    revised_rates = pd.Series(index=ages)\n",
+    "    revised_rates.index.name = 'age'\n",
     "    for a in ages:\n",
     "        idx = next((i for i, age in enumerate(rates.index) if age > a), 0)  # The zero is a trick to get the last element when one is subtracted\n",
     "        revised_rates.loc[a] = rates.iloc[idx - 1]\n",
@@ -74,7 +75,7 @@
     "    cohort_size = 100000\n",
     "    for a in rates.index:\n",
     "        lifetable[a] = cohort_size\n",
-    "        cohort_size -= cohort_size * age_rates.loc[a]\n",
+    "        cohort_size -= cohort_size * rates.loc[a]\n",
     "    return lifetable"
    ]
   },
@@ -85,7 +86,7 @@
    "outputs": [],
    "source": [
     "all_rates = pd.DataFrame(columns=years)\n",
-    "lifetables = pd.DataFrame(columns=years)\n",
+    "lifetables = pd.DataFrame(columns=all_rates.columns)\n",
     "for year in years:\n",
     "    all_rates[year] = adapt_death_rates_for_lifetable(100, raw_death_rates[year])\n",
     "    lifetables[year] = get_lifetable_from_rates(all_rates[year])"
@@ -97,7 +98,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "lifetable.plot(labels={'value': 'survivng', 'index': 'age'}, height=500, title='life table').update_layout(showlegend=False)"
+    "lifetables.plot(labels={'value': 'surviving'}, height=600, title='life tables by year')"
    ]
   }
  ],