c

pola-rs · Sep 27, 2024 · e4af44f · e4af44f
1 parent f1508b5
commit e4af44f
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/crates/polars-expr/src/expressions/window.rs b/crates/polars-expr/src/expressions/window.rs
@@ -335,7 +335,11 @@ impl WindowExpr {
             (WindowMapping::GroupsToRows, AggState::AggregatedList(_))
                 if gb.get_groups().is_sorted_flag() =>
             {
-                Ok(MapStrategy::Explode)
+                if let GroupsProxy::Idx(_) = gb.get_groups() {
+                    Ok(MapStrategy::Map)
+                } else {
+                    Ok(MapStrategy::Explode)
+                }
             },
             (WindowMapping::GroupsToRows, AggState::AggregatedList(_)) => Ok(MapStrategy::Map),
             // no aggregations, just return column