polars_plan/dsl/
cat.rs

1use super::*;
2
3/// Specialized expressions for Categorical dtypes.
4pub struct CategoricalNameSpace(pub(crate) Expr);
5
6impl CategoricalNameSpace {
7    pub fn get_categories(self) -> Expr {
8        self.0
9            .apply_private(CategoricalFunction::GetCategories.into())
10    }
11
12    #[cfg(feature = "strings")]
13    pub fn len_bytes(self) -> Expr {
14        self.0
15            .map_private(FunctionExpr::Categorical(CategoricalFunction::LenBytes))
16    }
17
18    #[cfg(feature = "strings")]
19    pub fn len_chars(self) -> Expr {
20        self.0
21            .map_private(FunctionExpr::Categorical(CategoricalFunction::LenChars))
22    }
23
24    #[cfg(feature = "strings")]
25    pub fn starts_with(self, prefix: String) -> Expr {
26        self.0
27            .map_private(FunctionExpr::Categorical(CategoricalFunction::StartsWith(
28                prefix,
29            )))
30    }
31
32    #[cfg(feature = "strings")]
33    pub fn ends_with(self, suffix: String) -> Expr {
34        self.0
35            .map_private(FunctionExpr::Categorical(CategoricalFunction::EndsWith(
36                suffix,
37            )))
38    }
39}
polars_plan/dsl/cat.rs

polars_plan/dsl/
cat.rs
