improve utf8 broadcasting

pola-rs · Nov 17, 2021 · de82bb1 · de82bb1
1 parent 8456594
commit de82bb1
Show file tree

Hide file tree

Showing 2 changed files with 36 additions and 1 deletion.
diff --git a/polars/polars-core/src/chunked_array/comparison.rs b/polars/polars-core/src/chunked_array/comparison.rs
@@ -459,6 +459,12 @@ impl ChunkCompare<&Utf8Chunked> for Utf8Chunked {
             } else {
                 BooleanChunked::full("", false, self.len())
             }
+        } else if self.len() == 1 {
+            if let Some(value) = self.get(0) {
+                rhs.eq(value)
+            } else {
+                BooleanChunked::full("", false, self.len())
+            }
         }
         // same length
         else if self.chunk_id().zip(rhs.chunk_id()).all(|(l, r)| l == r) {
@@ -476,6 +482,12 @@ impl ChunkCompare<&Utf8Chunked> for Utf8Chunked {
             } else {
                 BooleanChunked::full("", false, self.len())
             }
+        } else if self.len() == 1 {
+            if let Some(value) = self.get(0) {
+                rhs.neq(value)
+            } else {
+                BooleanChunked::full("", false, self.len())
+            }
         }
         // same length
         else if self.chunk_id().zip(rhs.chunk_id()).all(|(l, r)| l == r) {
@@ -493,6 +505,12 @@ impl ChunkCompare<&Utf8Chunked> for Utf8Chunked {
             } else {
                 BooleanChunked::full("", false, self.len())
             }
+        } else if self.len() == 1 {
+            if let Some(value) = self.get(0) {
+                rhs.lt(value)
+            } else {
+                BooleanChunked::full("", false, self.len())
+            }
         }
         // same length
         else if self.chunk_id().zip(rhs.chunk_id()).all(|(l, r)| l == r) {
@@ -510,6 +528,12 @@ impl ChunkCompare<&Utf8Chunked> for Utf8Chunked {
             } else {
                 BooleanChunked::full("", false, self.len())
             }
+        } else if self.len() == 1 {
+            if let Some(value) = self.get(0) {
+                rhs.lt_eq(value)
+            } else {
+                BooleanChunked::full("", false, self.len())
+            }
         }
         // same length
         else if self.chunk_id().zip(rhs.chunk_id()).all(|(l, r)| l == r) {
@@ -527,6 +551,12 @@ impl ChunkCompare<&Utf8Chunked> for Utf8Chunked {
             } else {
                 BooleanChunked::full("", false, self.len())
             }
+        } else if self.len() == 1 {
+            if let Some(value) = self.get(0) {
+                rhs.gt(value)
+            } else {
+                BooleanChunked::full("", false, self.len())
+            }
         }
         // same length
         else if self.chunk_id().zip(rhs.chunk_id()).all(|(l, r)| l == r) {
@@ -544,6 +574,12 @@ impl ChunkCompare<&Utf8Chunked> for Utf8Chunked {
             } else {
                 BooleanChunked::full("", false, self.len())
             }
+        } else if self.len() == 1 {
+            if let Some(value) = self.get(0) {
+                rhs.gt_eq(value)
+            } else {
+                BooleanChunked::full("", false, self.len())
+            }
         }
         // same length
         else if self.chunk_id().zip(rhs.chunk_id()).all(|(l, r)| l == r) {

diff --git a/polars/polars-core/src/utils/mod.rs b/polars/polars-core/src/utils/mod.rs
@@ -885,7 +885,6 @@ pub(crate) unsafe fn copy_from_slice_unchecked<T>(src: &[T], dst: &mut [T]) {
 #[cfg(test)]
 mod test {
     use super::*;
-    use crate::prelude::*;
 
     #[test]
     fn test_align_chunks() {