Fix stat_summary_bin when using binwidth & breaks

has2k1 · Aug 1, 2024 · 7a25a92 · 7a25a92
1 parent 29c77f3
commit 7a25a92
Show file tree

Hide file tree

Showing 4 changed files with 9 additions and 0 deletions.
diff --git a/doc/changelog.qmd b/doc/changelog.qmd
@@ -48,6 +48,9 @@ title: Changelog
 - Fixed bug in [](:class:`~plotnine.geoms.geom_segment`) where the lineend
   parameter was ignored. ({{< issue 727 >}})
 
+- Fixed [](:class:`~plotnine.stat_summary_bin`) to work with continuous `x`
+  aesthetic when specifying the `binwidth` or `breaks`. ({{< issue 824 >}})
+
 ## v0.13.6
 (2024-05-09)
 

diff --git a/plotnine/stats/stat_summary_bin.py b/plotnine/stats/stat_summary_bin.py
@@ -141,6 +141,7 @@ def compute_group(cls, data, scales, **params):
         )
 
         breaks = fuzzybreaks(scales.x, breaks, boundary, binwidth, bins)
+        bins = len(breaks) - 1
         data["bin"] = pd.cut(
             data["x"],
             bins=breaks,  # pyright: ignore

diff --git a/tests/baseline_images/test_stat_summary_bin/setting_binwidth.png b/tests/baseline_images/test_stat_summary_bin/setting_binwidth.png
diff --git a/tests/test_stat_summary_bin.py b/tests/test_stat_summary_bin.py
@@ -26,3 +26,8 @@ def test_continuous_x():
     )
 
     assert p == "continuous_x"
+
+
+def test_setting_binwidth():
+    p = ggplot(data, aes("xc", "y")) + stat_summary_bin(binwidth=3, geom="bar")
+    assert p == "setting_binwidth"