Add fast quit for numeric

pola-rs · Jan 3, 2025 · 18c61be · 18c61be
1 parent de6240e
commit 18c61be
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/crates/polars-ops/src/series/ops/horizontal.rs b/crates/polars-ops/src/series/ops/horizontal.rs
@@ -350,7 +350,10 @@ pub fn mean_horizontal(
         // All remaining must be numeric (or null).
         for col in &columns[first_non_null_idx + 1..] {
             let dtype = col.dtype();
-            if !(dtype.is_numeric()
+            if !ignore_nulls && dtype == &DataType::Null {
+                // The presence of a single null column guarantees the output is all-null.
+                return null_with_supertype(columns, true);
+            } else if !(dtype.is_numeric()
                 || dtype.is_decimal()
                 || dtype.is_bool()
                 || dtype.is_temporal()