
    Lpj*                       S SK Jr  S SKJrJrJr  S SKJrJr  S SK	J
r
  S SKJr  \(       a  S SKJr  S SKJr  S SKJr  S S	KJr   " S
 S\5      rSS jrSS jrSS jrSS jrSS jrSS jrSS jrSS jr  S     SS jjr/ SQrg)    )annotations)TYPE_CHECKINGAnyNoReturn)ExprKindExprNode)flatten)Expr)Iterable)timezone)DType)TimeUnitc                  Z    \ rS rSrSS jrSS jrSS jrSS jrSS jrSS jr	SS jr
S	rg
)Selector   c                &    [        U R                  6 $ N)r
   _nodes)selfs    N/var/www/html/pdf-tiff/venv/lib/python3.13/site-packages/narwhals/selectors.py_to_exprSelector._to_expr   s    T[[!!    c           	         [        U[        5      (       a  Sn[        U5      eU R                  5       R	                  [        [        R                  SU4SS95      $ )Nz=unsupported operand type(s) for op: ('Selector' + 'Selector')__add__Texprs
str_as_lit)
isinstancer   	TypeErrorr   _append_noder   r   ELEMENTWISE)r   othermsgs      r   r   Selector.__add__   sN    eX&&QCC. }}++X))9UHQUV
 	
r   c           
         [        U[        5      (       a+  U R                  [        [        R
                  SU4SSS95      $ U R                  5       R                  [        [        R
                  SU4SS95      $ )N__or__Tr   r   allow_multi_outputr   r   r   r!   r   r   r"   r   r   r#   s     r   r'   Selector.__or__   ss    eX&&$$(( (#'+  }}++X))8E8PTU
 	
r   c           
         [        U[        5      (       a+  U R                  [        [        R
                  SU4SSS95      $ U R                  5       R                  [        [        R
                  SU4SS95      $ )N__and__Tr(   r   r*   r+   s     r   r.   Selector.__and__,   ss    eX&&$$(( (#'+  }}++X))9UHQUV
 	
r   c                    [         er   NotImplementedErrorr+   s     r   __rsub__Selector.__rsub__;       !!r   c                    [         er   r1   r+   s     r   __rand__Selector.__rand__>   r5   r   c                    [         er   r1   r+   s     r   __ror__Selector.__ror__A   r5   r    N)returnr
   )r#   r   r=   r
   )r#   r   r=   r   )__name__
__module____qualname____firstlineno__r   r   r'   r.   r3   r7   r:   __static_attributes__r<   r   r   r   r      s%    "


"""r   r   c                 \    [        U 5      n[        [        [        R                  SUS95      $ )a&  Select columns based on their dtype.

Arguments:
    dtypes: one or data types to select

Examples:
    >>> import pyarrow as pa
    >>> import narwhals as nw
    >>> import narwhals.selectors as ncs
    >>> df_native = pa.table({"a": [1, 2], "b": ["x", "y"], "c": [4.1, 2.3]})
    >>> df = nw.from_native(df_native)

    Let's select int64 and float64  dtypes and multiply each value by 2:

    >>> df.select(ncs.by_dtype(nw.Int64, nw.Float64) * 2).to_native()
    pyarrow.Table
    a: int64
    c: double
    ----
    a: [[2,4]]
    c: [[8.2,4.6]]
zselectors.by_dtype)dtypes)r	   r   r   r   SELECTOR)rD   	flatteneds     r   by_dtyperG   E   s(    . IHX..0DYWXXr   c                F    [        [        [        R                  SU S95      $ )a7  Select all columns that match the given regex pattern.

Arguments:
    pattern: A valid regular expression pattern.

Examples:
    >>> import pandas as pd
    >>> import narwhals as nw
    >>> import narwhals.selectors as ncs
    >>> df_native = pd.DataFrame(
    ...     {"bar": [123, 456], "baz": [2.0, 5.5], "zap": [0, 1]}
    ... )
    >>> df = nw.from_native(df_native)

    Let's select column names containing an 'a', preceded by a character that is not 'z':

    >>> df.select(ncs.matches("[^z]a")).to_native()
       bar  baz
    0  123  2.0
    1  456  5.5
zselectors.matchespatternr   r   r   rE   rI   s    r   matchesrL   `   s    , HX..0CWUVVr   c                 H    [        [        [        R                  S5      5      $ )u~  Select numeric columns.

Examples:
    >>> import polars as pl
    >>> import narwhals as nw
    >>> import narwhals.selectors as ncs
    >>> df_native = pl.DataFrame({"a": [1, 2], "b": ["x", "y"], "c": [4.1, 2.3]})
    >>> df = nw.from_native(df_native)

    Let's select numeric dtypes and multiply each value by 2:

    >>> df.select(ncs.numeric() * 2).to_native()
    shape: (2, 2)
    ┌─────┬─────┐
    │ a   ┆ c   │
    │ --- ┆ --- │
    │ i64 ┆ f64 │
    ╞═════╪═════╡
    │ 2   ┆ 8.2 │
    │ 4   ┆ 4.6 │
    └─────┴─────┘
zselectors.numericrK   r<   r   r   numericrN   y   s    . HX..0CDEEr   c                 H    [        [        [        R                  S5      5      $ )u%  Select boolean columns.

Examples:
    >>> import polars as pl
    >>> import narwhals as nw
    >>> import narwhals.selectors as ncs
    >>> df_native = pl.DataFrame({"a": [1, 2], "b": ["x", "y"], "c": [False, True]})
    >>> df = nw.from_native(df_native)

    Let's select boolean dtypes:

    >>> df.select(ncs.boolean())
    ┌──────────────────┐
    |Narwhals DataFrame|
    |------------------|
    |  shape: (2, 1)   |
    |  ┌───────┐       |
    |  │ c     │       |
    |  │ ---   │       |
    |  │ bool  │       |
    |  ╞═══════╡       |
    |  │ false │       |
    |  │ true  │       |
    |  └───────┘       |
    └──────────────────┘
zselectors.booleanrK   r<   r   r   booleanrP      s    6 HX..0CDEEr   c                 H    [        [        [        R                  S5      5      $ )u  Select string columns.

Examples:
    >>> import polars as pl
    >>> import narwhals as nw
    >>> import narwhals.selectors as ncs
    >>> df_native = pl.DataFrame({"a": [1, 2], "b": ["x", "y"], "c": [False, True]})
    >>> df = nw.from_native(df_native)

    Let's select string dtypes:

    >>> df.select(ncs.string()).to_native()
    shape: (2, 1)
    ┌─────┐
    │ b   │
    │ --- │
    │ str │
    ╞═════╡
    │ x   │
    │ y   │
    └─────┘
zselectors.stringrK   r<   r   r   stringrR      s    . HX..0BCDDr   c                 H    [        [        [        R                  S5      5      $ )u  Select categorical columns.

Examples:
    >>> import polars as pl
    >>> import narwhals as nw
    >>> import narwhals.selectors as ncs
    >>> df_native = pl.DataFrame({"a": [1, 2], "b": ["x", "y"], "c": [False, True]})

    Let's convert column "b" to categorical, and then select categorical dtypes:

    >>> df = nw.from_native(df_native).with_columns(
    ...     b=nw.col("b").cast(nw.Categorical())
    ... )
    >>> df.select(ncs.categorical()).to_native()
    shape: (2, 1)
    ┌─────┐
    │ b   │
    │ --- │
    │ cat │
    ╞═════╡
    │ x   │
    │ y   │
    └─────┘
zselectors.categoricalrK   r<   r   r   categoricalrT      s    2 HX..0GHIIr   c                 H    [        [        [        R                  S5      5      $ )uC  Select enum columns.

Examples:
    >>> import polars as pl
    >>> import narwhals as nw
    >>> import narwhals.selectors as ncs
    >>> df_native = pl.DataFrame(
    ...     {"a": [1, 2], "b": ["x", "y"]},
    ...     schema_overrides={"b": pl.Enum(["x", "y"])},
    ... )
    >>> df = nw.from_native(df_native)

    Let's select enum dtypes:

    >>> df.select(ncs.enum()).to_native()
    shape: (2, 1)
    ┌──────┐
    │ b    │
    │ ---  │
    │ enum │
    ╞══════╡
    │ x    │
    │ y    │
    └──────┘
zselectors.enumrK   r<   r   r   enumrV      s    4 HX..0@ABBr   c                 H    [        [        [        R                  S5      5      $ )a  Select all columns.

Examples:
    >>> import pandas as pd
    >>> import narwhals as nw
    >>> import narwhals.selectors as ncs
    >>> df_native = pd.DataFrame({"a": [1, 2], "b": ["x", "y"], "c": [False, True]})
    >>> df = nw.from_native(df_native)

    Let's select all dtypes:

    >>> df.select(ncs.all()).to_native()
       a  b      c
    0  1  x  False
    1  2  y   True
zselectors.allrK   r<   r   r   allrX     s    " HX..@AAr   Nc           	     H    [        [        [        R                  SU US95      $ )aA  Select all datetime columns, optionally filtering by time unit/zone.

Arguments:
    time_unit: One (or more) of the allowed timeunit precision strings, "ms", "us",
        "ns" and "s". Omit to select columns with any valid timeunit.
    time_zone: Specify which timezone(s) to select

        * One or more timezone strings, as defined in zoneinfo (to see valid options
            run `import zoneinfo; zoneinfo.available_timezones()` for a full list).
        * Set `None` to select Datetime columns that do not have a timezone.
        * Set `"*"` to select Datetime columns that have *any* timezone.

Examples:
    >>> from datetime import datetime, timezone
    >>> import pyarrow as pa
    >>> import narwhals as nw
    >>> import narwhals.selectors as ncs
    >>>
    >>> utc_tz = timezone.utc
    >>> data = {
    ...     "tstamp_utc": [
    ...         datetime(2023, 4, 10, 12, 14, 16, 999000, tzinfo=utc_tz),
    ...         datetime(2025, 8, 25, 14, 18, 22, 666000, tzinfo=utc_tz),
    ...     ],
    ...     "tstamp": [
    ...         datetime(2000, 11, 20, 18, 12, 16, 600000),
    ...         datetime(2020, 10, 30, 10, 20, 25, 123000),
    ...     ],
    ...     "numeric": [3.14, 6.28],
    ... }
    >>> df_native = pa.table(data)
    >>> df_nw = nw.from_native(df_native)
    >>> df_nw.select(ncs.datetime()).to_native()
    pyarrow.Table
    tstamp_utc: timestamp[us, tz=UTC]
    tstamp: timestamp[us]
    ----
    tstamp_utc: [[2023-04-10 12:14:16.999000Z,2025-08-25 14:18:22.666000Z]]
    tstamp: [[2000-11-20 18:12:16.600000,2020-10-30 10:20:25.123000]]

    Select only datetime columns that have any time_zone specification:

    >>> df_nw.select(ncs.datetime(time_zone="*")).to_native()
    pyarrow.Table
    tstamp_utc: timestamp[us, tz=UTC]
    ----
    tstamp_utc: [[2023-04-10 12:14:16.999000Z,2025-08-25 14:18:22.666000Z]]
zselectors.datetime	time_unit	time_zonerK   rZ   s     r   datetimer]     s,    h  		
 r   )	rX   rP   rG   rT   r]   rV   rL   rN   rR   )rD   z3DType | type[DType] | Iterable[DType | type[DType]]r=   r   )rJ   strr=   r   )r=   r   )N)*N)r[   z$TimeUnit | Iterable[TimeUnit] | Noner\   z7str | timezone | Iterable[str | timezone | None] | Noner=   r   )
__future__r   typingr   r   r   narwhals._expression_parsingr   r   narwhals._utilsr	   narwhals.exprr
   collections.abcr   r]   r   narwhals.dtypesr   narwhals.typingr   r   rG   rL   rN   rP   rR   rT   rV   rX   __all__r<   r   r   <module>ri      s    " / / ; # (!%(1"t 1"hY6W2F4F<E4J8C:B* 7;IT;3;F; ;|
r   