Check if length of index in pivot operation is non-zero. Fixes: pola-…

…rs#3343. (pola-rs#3346)
ghuls · May 10, 2022 · e37620d · e37620d
1 parent 4c3644a
commit e37620d
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/polars/polars-core/src/frame/groupby/pivot.rs b/polars/polars-core/src/frame/groupby/pivot.rs
@@ -222,6 +222,12 @@ impl DataFrame {
         sort_columns: bool,
         stable: bool,
     ) -> Result<DataFrame> {
+        if index.is_empty() {
+            return Err(PolarsError::ComputeError(
+                "index cannot be zero length".into(),
+            ));
+        }
+
         let mut final_cols = vec![];
 
         let mut count = 0;