plot dynamics

calvin-zcx · Mar 10, 2023 · 7c9bc9f · 7c9bc9f
1 parent b1613ee
commit 7c9bc9f
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/iptw/screen_dx_recover_pregnancy.py b/iptw/screen_dx_recover_pregnancy.py
@@ -296,6 +296,9 @@ def select_subpopulation(df, severity):
     df = pd.concat(df_list, ignore_index=True)
     df = select_subpopulation(df, args.severity)
 
+    # df.to_csv('preg_pos_neg.csv')
+    # sys.exit(-1)
+
     # 'T2D-Obesity', 'Hypertension', 'Mental-substance', 'Corticosteroids'
     print('Severity cohorts:', args.severity, 'df.shape:', df.shape)
     col_names = pd.Series(df.columns)