docs: Update group_by_dynamic example

pola-rs · Oct 14, 2023 · e739d13 · e739d13
1 parent 55c9f8f
commit e739d13
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 12 deletions.
diff --git a/docs/src/python/user-guide/transformations/time-series/rolling.py b/docs/src/python/user-guide/transformations/time-series/rolling.py
@@ -29,17 +29,11 @@
     .to_frame()
 )
 
-out = (
-    df.group_by_dynamic("time", every="1mo", period="1mo", closed="left")
-    .agg(
-        [
-            pl.col("time").cumcount().reverse().head(3).alias("day/eom"),
-            ((pl.col("time") - pl.col("time").first()).last().dt.days() + 1).alias(
-                "days_in_month"
-            ),
-        ]
-    )
-    .explode("day/eom")
+out = df.group_by_dynamic("time", every="1mo", period="1mo", closed="left").agg(
+    pl.col("time").cumcount().reverse().head(3).alias("day/eom"),
+    ((pl.col("time") - pl.col("time").first()).last().dt.days() + 1).alias(
+        "days_in_month"
+    ),
 )
 print(out)
 # --8<-- [end:group_by_dyn]

diff --git a/docs/src/rust/user-guide/transformations/time-series/rolling.rs b/docs/src/rust/user-guide/transformations/time-series/rolling.rs
@@ -86,7 +86,6 @@ fn main() -> Result<(), Box<dyn std::error::Error>> {
             ) + lit(1))
             .alias("days_in_month"),
         ])
-        .explode([col("day/eom")])
         .collect()?;
     println!("{}", &out);
     // --8<-- [end:group_by_dyn]