update for new pandas

PSLmodels · Feb 4, 2024 · 055c6ee · 055c6ee
1 parent 44cb97b
commit 055c6ee
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/taxcalc/utils.py b/taxcalc/utils.py
@@ -261,6 +261,7 @@ def add_income_table_row_variable(dframe, income_measure, bin_edges):
     assert isinstance(bin_edges, list)
     dframe['table_row'] = pd.cut(dframe[income_measure],
                                  bin_edges, right=False)
+
     return dframe
 
 
@@ -539,8 +540,7 @@ def count_gt_zero(dframe, col_name, tolerance=0.001):
             df2, baseline_expanded_income, SOI_AGI_BINS)
     del df2
     # create grouped Pandas DataFrame
-    gdf = dframe.groupby('table_row', as_index=False)
-    del dframe['table_row']
+    gdf = dframe.groupby('table_row', as_index=False, observed=False)
     # create additive difference table statistics from gdf
     diff_table = additive_stats_dataframe(gdf)
     # calculate additive statistics on sums row