polars-plan 0.55.1

Lazy query engine for the Polars DataFrame library
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44

use super::*;

/// Specialized expressions for Categorical dtypes.
pub struct CategoricalNameSpace(pub(crate) Expr);

impl CategoricalNameSpace {
    pub fn get_categories(self) -> Expr {
        self.0.map_unary(CategoricalFunction::GetCategories)
    }

    #[cfg(feature = "strings")]
    pub fn len_bytes(self) -> Expr {
        self.0.map_unary(CategoricalFunction::LenBytes)
    }

    #[cfg(feature = "strings")]
    pub fn len_chars(self) -> Expr {
        self.0.map_unary(CategoricalFunction::LenChars)
    }

    #[cfg(feature = "strings")]
    pub fn starts_with(self, prefix: String) -> Expr {
        self.0.map_unary(CategoricalFunction::StartsWith(prefix))
    }

    #[cfg(feature = "strings")]
    pub fn ends_with(self, suffix: String) -> Expr {
        self.0.map_unary(CategoricalFunction::EndsWith(suffix))
    }

    #[cfg(feature = "strings")]
    pub fn slice(self, offset: i64, length: Option<usize>) -> Expr {
        self.0.map_unary(CategoricalFunction::Slice(offset, length))
    }

    pub fn to(self, dtype: impl Into<DataTypeExpr>, strict: bool) -> Expr {
        self.0
            .map_unary(CategoricalFunction::To(dtype.into(), strict))
    }

    pub fn physical(self) -> Expr {
        self.0.map_unary(CategoricalFunction::Physical)
    }
}