perf(rust, python): first check rev-map on categorical equality check (…

…pola-rs#6085)

polars/polars-core/src/chunked_array/logical/categorical/builder.rs

@@ -95,6 +95,16 @@ impl RevMapping {
         }
     }
 
+    pub fn get_optional(&self, idx: u32) -> Option<&str> {
+        match self {
+            Self::Global(map, a, _) => {
+                let idx = *map.get(&idx)?;
+                a.get(idx as usize)
+            }
+            Self::Local(a) => a.get(idx as usize),
+        }
+    }
+
     /// Categorical to str
     ///
     /// # Safety

polars/polars-core/src/series/comparison.rs

@@ -158,10 +158,17 @@ impl ChunkCompare<&Series> for Series {
             #[cfg(feature = "dtype-categorical")]
             (Categorical(Some(rev_map_l)), Categorical(Some(rev_map_r)), _, _) => {
                 if rev_map_l.same_src(rev_map_r) {
-                    self.categorical()
-                        .unwrap()
-                        .logical()
-                        .equal(rhs.categorical().unwrap().logical())
+                    let rhs = rhs.categorical().unwrap().logical();
+
+                    // first check the rev-map
+                    if rhs.len() == 1 && rhs.null_count() == 0 {
+                        let rhs = rhs.get(0).unwrap();
+                        if rev_map_l.get_optional(rhs).is_none() {
+                            return Ok(BooleanChunked::full(self.name(), false, self.len()));
+                        }
+                    }
+
+                    self.categorical().unwrap().logical().equal(rhs)
                 } else {
                     return Err(PolarsError::ComputeError("Cannot compare categoricals originating from different sources. Consider setting a global string cache.".into()));
                 }
@@ -203,10 +210,17 @@ impl ChunkCompare<&Series> for Series {
             #[cfg(feature = "dtype-categorical")]
             (Categorical(Some(rev_map_l)), Categorical(Some(rev_map_r)), _, _) => {
                 if rev_map_l.same_src(rev_map_r) {
-                    self.categorical()
-                        .unwrap()
-                        .logical()
-                        .not_equal(rhs.categorical().unwrap().logical())
+                    let rhs = rhs.categorical().unwrap().logical();
+
+                    // first check the rev-map
+                    if rhs.len() == 1 && rhs.null_count() == 0 {
+                        let rhs = rhs.get(0).unwrap();
+                        if rev_map_l.get_optional(rhs).is_none() {
+                            return Ok(BooleanChunked::full(self.name(), true, self.len()));
+                        }
+                    }
+
+                    self.categorical().unwrap().logical().not_equal(rhs)
                 } else {
                     return Err(PolarsError::ComputeError("Cannot compare categoricals originating from different sources. Consider setting a global string cache.".into()));
                 }