replace agg("count") with .count() --> ~0.1ms speedup

lux-org · Nov 11, 2020 · 08d167e · 08d167e
1 parent b6a7dd6
commit 08d167e
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/lux/executor/PandasExecutor.py b/lux/executor/PandasExecutor.py
@@ -358,9 +358,7 @@ def execute_2D_binning(vis: Vis):
                 result = result.dropna()
             else:
                 groups = vis._vis_data.groupby(["xBin", "yBin"])[x_attr]
-                result = groups.agg("count").reset_index(
-                    name=x_attr
-                )  # .agg in this line throws SettingWithCopyWarning
+                result = groups.count().reset_index(name=x_attr)  
                 result = result.rename(columns={x_attr: "count"})
                 result = result[result["count"] != 0]