diff options
Diffstat (limited to 'python/notebooks')
| -rw-r--r-- | python/notebooks/risk_sabo.ipynb | 42 |
1 files changed, 5 insertions, 37 deletions
diff --git a/python/notebooks/risk_sabo.ipynb b/python/notebooks/risk_sabo.ipynb index ccfea2bc..5106056c 100644 --- a/python/notebooks/risk_sabo.ipynb +++ b/python/notebooks/risk_sabo.ipynb @@ -3,11 +3,7 @@ { "cell_type": "code", "execution_count": null, - "metadata": { - "jupyter": { - "source_hidden": true - } - }, + "metadata": {}, "outputs": [], "source": [ "import datetime\n", @@ -96,37 +92,7 @@ "from utils.db import serenitas_pool\n", "conn = serenitas_pool.getconn()\n", "surv_curves = load_all_curves(conn, spread_date)\n", - "serenitas_pool.putconn(conn)\n", - "surv_prob = surv_curves['curve'].apply(lambda sc: pd.Series(sc.survival_probability(jtd.columns.date), index=jtd.columns))\n", - "surv_prob = surv_prob.loc[jtd.index]" - ] - }, - { - "cell_type": "code", - "execution_count": null, - "metadata": {}, - "outputs": [], - "source": [ - "jtd = (1-surv_prob) * jtd\n", - "jtd = jtd.groupby(jtd.index.str.extract(\"([^_]*)\", expand=False)).sum()" - ] - }, - { - "cell_type": "code", - "execution_count": null, - "metadata": {}, - "outputs": [], - "source": [ - "jtd" - ] - }, - { - "cell_type": "code", - "execution_count": null, - "metadata": {}, - "outputs": [], - "source": [ - "ref_names = surv_curves.groupby(surv_curves.index.str.extract(\"([^_]*)\", expand=False))['name'].first()" + "serenitas_pool.putconn(conn)" ] }, { @@ -135,7 +101,9 @@ "metadata": {}, "outputs": [], "source": [ - "ref_names[jtd.index]" + "jtd_sabo = jtd[[jtd.columns[0]]].join(surv_curves.groupby(level=0).first()[['name', 'company_id']])\n", + "jtd_sabo.columns = ['jtd', 'name', 'company_id']\n", + "jtd_sabo.to_csv(base_dir / f\"jtd_{position_date:%Y%m%d}.csv\")" ] }, { |