Update gd_merge.py

luna983 · Feb 6, 2021 · b9b868c · b9b868c
1 parent 4325bd4
commit b9b868c
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/scripts/gd_merge.py b/scripts/gd_merge.py
@@ -123,8 +123,10 @@
         df_draw = df_draw.astype(int)
         df = pd.merge(df, df_draw, how='left', on=['grid_lon', 'grid_lat'])
 
-    # check treat_eligible distribution
-    # print('treat_eligible: \n', df['treat_eligible'].value_counts())
+    # check variable distribution
+    print('treat_eligible: \n', df['treat_eligible'].value_counts())
+    print('eligible: \n', df['eligible'].value_counts())
+    print('n_household: \n', df['n_household'].value_counts())
     # print metadata
     print('>>> df.shape')
     print(df.shape)