include time mass frac

Author: PabloCorcho

tutorials/models/analytical.ipynb

@@ -166,6 +166,11 @@
     "sfr = np.diff(mfh) / np.diff(cosmic_time)\n",
     "t_sfr = (cosmic_time[1:] + cosmic_time[:-1]) / 2\n",
     "\n",
+    "# Estimate the times at which the model formed a given fraction of the present\n",
+    "# mass\n",
+    "mass_fractions = [0.1, 0.5, 0.9, 0.99]\n",
+    "times_frac = model.time_at_stellar_mass_frac(mass_fractions)\n",
+    "\n",
     "fig, axes = plt.subplots(nrows=3, sharex=True)\n",
     "fig.suptitle(\"Log-normal SFH with a quenching event\")\n",
     "fig.supxlabel(f\"cosmic time ({cosmic_time.unit})\")\n",
@@ -175,6 +180,12 @@
     "ax.set_yscale('log')\n",
     "ax.plot(cosmic_time, mfh.to_value(u.Msun), '.-')\n",
     "\n",
+    "for t, m in zip(times_frac, mass_fractions):\n",
+    "    ax.annotate(f\"{m * 100:.1f}%\", xy=(t.value, 1e9), va=\"top\", ha=\"center\",\n",
+    "                arrowprops=dict(facecolor='black', shrink=0.05))\n",
+    "    # ax.arrow(t.value, 1e9, dx=0, dy=3e9, width=0.03,\n",
+    "    #          head_width=0.1, head_length=0.1)\n",
+    "\n",
     "ax = axes[1]\n",
     "ax.set_ylabel(r\"SFR (M$_\\odot$/yr)\")\n",
     "ax.set_yscale('log')\n",
@@ -183,7 +194,7 @@
     "ax = axes[2]\n",
     "ax.set_ylabel(r\"Z\")\n",
     "ax.set_yscale('log')\n",
-    "ax.plot(cosmic_time, model.ism_metallicity(cosmic_time), '.-')\n",
+    "ax.plot(cosmic_time, model.ism_metallicity(cosmic_time), '.-')  \n",
     "\n",
     "for ax in axes:\n",
     "    ax.axvline(quenching_time.to_value(cosmic_time.unit), color='r', label=\"Quenching event\")\n",
@@ -243,7 +254,7 @@
     "ax.set_yscale('log')\n",
     "ax.set_ylim(5e-5, 0.05)\n",
     "\n",
-    "mappable = ax.pcolormesh(ssp.ages.to_value(\"Gyr\"), ssp.metallicities, ssp_weights.to_value(u.Msun), norm=LogNorm())\n",
+    "mappable = ax.pcolormesh(ssp.ages.to_value(\"Gyr\"), ssp.metallicities.value, ssp_weights.to_value(u.Msun), norm=LogNorm())\n",
     "\n",
     "plt.colorbar(mappable, ax=ax, label=r\"SSP mass (M$_\\odot$)\")\n",
     "\n",

tutorials/models/particles.ipynb

@@ -183,8 +183,10 @@
     "ax.set_yscale('log')\n",
     "ax.set_ylim(5e-5, 0.05)\n",
     "\n",
-    "mappable = ax.pcolormesh(ssp.ages.to_value(\"Gyr\"), ssp.metallicities, ssp_weights, norm=LogNorm())\n",
-    "ax.plot((today - particles_t_form).to_value(u.Gyr), particles_z, 'k,', alpha=.25)\n",
+    "mappable = ax.pcolormesh(ssp.ages.to_value(\"Gyr\"), ssp.metallicities.value,\n",
+    "                         ssp_weights.value, norm=LogNorm())\n",
+    "ax.plot((today - particles_t_form).to_value(u.Gyr), particles_z,\n",
+    "        'k,', alpha=.25)\n",
     "\n",
     "plt.colorbar(mappable, ax=ax, label=r\"SSP mass (M$_\\odot$)\")"
    ]
@@ -226,9 +228,9 @@
  ],
  "metadata": {
   "kernelspec": {
-   "display_name": "test-pst",
+   "display_name": "base",
    "language": "python",
-   "name": "test-pst"
+   "name": "python3"
   },
   "language_info": {
    "codemirror_mode": {
@@ -240,7 +242,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.8.10"
+   "version": "3.11.5"
   }
  },
  "nbformat": 4,