§
    }Àûi”7  ã                  ó”   — d dl mZ d dlmZmZmZ d dlmZmZ erd dl	m
Z
 d dlmZ  edd¬¦  «        Z G d	„ d
ee         ¦  «        ZdS )é    )Úannotations)ÚTYPE_CHECKINGÚGenericÚTypeVar)ÚExprKindÚExprNode)ÚExpr)ÚNonNestedLiteralÚExprTr	   )Úboundc                  ón   — e Zd Zdd„Zdd„Zdd„Zdd
„Zdd„Zdd„Zdd„Z	dd„Z
dd„Zdd„Zdddœdd„ZdS )ÚExprListNamespaceÚexprr   ÚreturnÚNonec                ó   — || _         d S )N)Ú_expr)Úselfr   s     úF/home/jrussi/.local/lib/python3.11/site-packages/narwhals/expr_list.pyÚ__init__zExprListNamespace.__init__   s   € ØˆŒ
ˆ
ˆ
ó    c                óf   — | j                              t          t          j        d¦  «        ¦  «        S )ué  Return the number of elements in each list.

        Null values count towards the total.

        Examples:
            >>> import polars as pl
            >>> import narwhals as nw
            >>> df_native = pl.DataFrame({"a": [[1, 2], [3, 4, None], None, []]})
            >>> df = nw.from_native(df_native)
            >>> df.with_columns(a_len=nw.col("a").list.len())
            â”Œâ”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”
            |   Narwhals DataFrame   |
            |------------------------|
            |shape: (4, 2)           |
            |â”Œâ”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”¬â”€â”€â”€â”€â”€â”€â”€â”|
            |â”‚ a            â”† a_len â”‚|
            |â”‚ ---          â”† ---   â”‚|
            |â”‚ list[i64]    â”† u32   â”‚|
            |â•žâ•â•â•â•â•â•â•â•â•â•â•â•â•â•â•ªâ•â•â•â•â•â•â•â•¡|
            |â”‚ [1, 2]       â”† 2     â”‚|
            |â”‚ [3, 4, null] â”† 3     â”‚|
            |â”‚ null         â”† null  â”‚|
            |â”‚ []           â”† 0     â”‚|
            |â””â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”´â”€â”€â”€â”€â”€â”€â”€â”˜|
            â””â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”˜
        zlist.len©r   Ú_append_noder   r   ÚELEMENTWISE©r   s    r   ÚlenzExprListNamespace.len   s'   € ð6 Œz×&Ò&¥xµÔ0DÀjÑ'QÔ'QÑRÔRÐRr   c                óf   — | j                              t          t          j        d¦  «        ¦  «        S )u‡  Get the unique/distinct values in the list.

        Null values are included in the result. The order of unique values is not guaranteed.

        Examples:
            >>> import polars as pl
            >>> import narwhals as nw
            >>> df_native = pl.DataFrame({"a": [[1, 1, 2], [3, 3, None], None, []]})
            >>> df = nw.from_native(df_native)
            >>> df.with_columns(a_unique=nw.col("a").list.unique())
            â”Œâ”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”
            |     Narwhals DataFrame     |
            |----------------------------|
            |shape: (4, 2)               |
            |â”Œâ”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”¬â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”|
            |â”‚ a            â”† a_unique  â”‚|
            |â”‚ ---          â”† ---       â”‚|
            |â”‚ list[i64]    â”† list[i64] â”‚|
            |â•žâ•â•â•â•â•â•â•â•â•â•â•â•â•â•â•ªâ•â•â•â•â•â•â•â•â•â•â•â•¡|
            |â”‚ [1, 1, 2]    â”† [1, 2]    â”‚|
            |â”‚ [3, 3, null] â”† [null, 3] â”‚|
            |â”‚ null         â”† null      â”‚|
            |â”‚ []           â”† []        â”‚|
            |â””â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”´â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”˜|
            â””â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”˜
        zlist.uniquer   r   s    r   ÚuniquezExprListNamespace.unique/   s'   € ð6 Œz×&Ò&¥xµÔ0DÀmÑ'TÔ'TÑUÔUÐUr   Úitemr
   c                ój   — | j                              t          t          j        d|¬¦  «        ¦  «        S )u;  Check if sublists contain the given item.

        Arguments:
            item: Item that will be checked for membership.

        Examples:
            >>> import polars as pl
            >>> import narwhals as nw
            >>> df_native = pl.DataFrame({"a": [[1, 2], None, []]})
            >>> df = nw.from_native(df_native)
            >>> df.with_columns(a_contains_1=nw.col("a").list.contains(1))
            â”Œâ”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”
            |     Narwhals DataFrame     |
            |----------------------------|
            |shape: (3, 2)               |
            |â”Œâ”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”¬â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”|
            |â”‚ a         â”† a_contains_1 â”‚|
            |â”‚ ---       â”† ---          â”‚|
            |â”‚ list[i64] â”† bool         â”‚|
            |â•žâ•â•â•â•â•â•â•â•â•â•â•â•ªâ•â•â•â•â•â•â•â•â•â•â•â•â•â•â•¡|
            |â”‚ [1, 2]    â”† true         â”‚|
            |â”‚ null      â”† null         â”‚|
            |â”‚ []        â”† false        â”‚|
            |â””â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”´â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”˜|
            â””â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”˜
        zlist.contains)r    r   )r   r    s     r   ÚcontainszExprListNamespace.containsL   s3   € ð6 Œz×&Ò&Ý•XÔ)¨?ÀÐFÑFÔFñ
ô 
ð 	
r   ÚindexÚintc                ó  — t          |t          ¦  «        s'dt          |¦  «        j        › d}t	          |¦  «        ‚|dk     rd|› d}t          |¦  «        ‚| j                             t          t          j
        d|¬¦  «        ¦  «        S )uê  Return the value by index in each list.

        Negative indices are not accepted.

        Examples:
            >>> import polars as pl
            >>> import narwhals as nw
            >>> df_native = pl.DataFrame({"a": [[1, 2], [3, 4, None], [None, 5]]})
            >>> df = nw.from_native(df_native)
            >>> df.with_columns(a_first=nw.col("a").list.get(0))
            â”Œâ”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”
            |    Narwhals DataFrame    |
            |--------------------------|
            |shape: (3, 2)             |
            |â”Œâ”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”¬â”€â”€â”€â”€â”€â”€â”€â”€â”€â”|
            |â”‚ a            â”† a_first â”‚|
            |â”‚ ---          â”† ---     â”‚|
            |â”‚ list[i64]    â”† i64     â”‚|
            |â•žâ•â•â•â•â•â•â•â•â•â•â•â•â•â•â•ªâ•â•â•â•â•â•â•â•â•â•¡|
            |â”‚ [1, 2]       â”† 1       â”‚|
            |â”‚ [3, 4, null] â”† 3       â”‚|
            |â”‚ [null, 5]    â”† null    â”‚|
            |â””â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”´â”€â”€â”€â”€â”€â”€â”€â”€â”€â”˜|
            â””â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”˜
        z'Index must be of type 'int'. Got type 'z
' instead.r   zIndex z8 is out of bounds: should be greater than or equal to 0.zlist.get)r#   )Ú
isinstancer$   ÚtypeÚ__name__Ú	TypeErrorÚ
ValueErrorr   r   r   r   r   )r   r#   Úmsgs      r   ÚgetzExprListNamespace.getk   s”   € õ4 ˜%¥Ñ%Ô%ð 	!àZ½$¸u¹+¼+Ô:NÐZÐZÐZð õ ˜C‘.”.Ð à1Š9ˆ9ØZ˜5ÐZÐZÐZˆCÝ˜S‘/”/Ð!àŒz×&Ò&Ý•XÔ)¨:¸UÐCÑCÔCñ
ô 
ð 	
r   c                óf   — | j                              t          t          j        d¦  «        ¦  «        S )u  Compute the min value of the lists in the array.

        Examples:
            >>> import duckdb
            >>> import narwhals as nw
            >>> df_native = duckdb.sql("SELECT * FROM VALUES ([1]), ([3, 4, NULL]) df(a)")
            >>> df = nw.from_native(df_native)
            >>> df.with_columns(a_min=nw.col("a").list.min())
            â”Œâ”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”
            |   Narwhals LazyFrame   |
            |------------------------|
            |â”Œâ”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”¬â”€â”€â”€â”€â”€â”€â”€â”|
            |â”‚      a       â”‚ a_min â”‚|
            |â”‚   int32[]    â”‚ int32 â”‚|
            |â”œâ”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”¼â”€â”€â”€â”€â”€â”€â”€â”¤|
            |â”‚ [1]          â”‚     1 â”‚|
            |â”‚ [3, 4, NULL] â”‚     3 â”‚|
            |â””â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”´â”€â”€â”€â”€â”€â”€â”€â”˜|
            â””â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”˜
        zlist.minr   r   s    r   ÚminzExprListNamespace.min“   s'   € ð* Œz×&Ò&¥xµÔ0DÀjÑ'QÔ'QÑRÔRÐRr   c                óf   — | j                              t          t          j        d¦  «        ¦  «        S )uY  Compute the max value of the lists in the array.

        Examples:
            >>> import polars as pl
            >>> import narwhals as nw
            >>> df_native = pl.DataFrame({"a": [[1], [3, 4, None]]})
            >>> df = nw.from_native(df_native)
            >>> df.with_columns(a_max=nw.col("a").list.max())
            â”Œâ”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”
            |   Narwhals DataFrame   |
            |------------------------|
            |shape: (2, 2)           |
            |â”Œâ”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”¬â”€â”€â”€â”€â”€â”€â”€â”|
            |â”‚ a            â”† a_max â”‚|
            |â”‚ ---          â”† ---   â”‚|
            |â”‚ list[i64]    â”† i64   â”‚|
            |â•žâ•â•â•â•â•â•â•â•â•â•â•â•â•â•â•ªâ•â•â•â•â•â•â•â•¡|
            |â”‚ [1]          â”† 1     â”‚|
            |â”‚ [3, 4, null] â”† 4     â”‚|
            |â””â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”´â”€â”€â”€â”€â”€â”€â”€â”˜|
            â””â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”˜
        zlist.maxr   r   s    r   ÚmaxzExprListNamespace.maxª   ó'   € ð. Œz×&Ò&¥xµÔ0DÀjÑ'QÔ'QÑRÔRÐRr   c                óf   — | j                              t          t          j        d¦  «        ¦  «        S )u?  Compute the mean value of the lists in the array.

        Examples:
            >>> import pyarrow as pa
            >>> import narwhals as nw
            >>> df_native = pa.table({"a": [[1], [3, 4, None]]})
            >>> df = nw.from_native(df_native)
            >>> df.with_columns(a_mean=nw.col("a").list.mean())
            â”Œâ”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”
            |  Narwhals DataFrame  |
            |----------------------|
            |pyarrow.Table         |
            |a: list<item: int64>  |
            |  child 0, item: int64|
            |a_mean: double        |
            |----                  |
            |a: [[[1],[3,4,null]]] |
            |a_mean: [[1,3.5]]     |
            â””â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”˜
        z	list.meanr   r   s    r   ÚmeanzExprListNamespace.meanÃ   s'   € ð* Œz×&Ò&¥xµÔ0DÀkÑ'RÔ'RÑSÔSÐSr   c                óf   — | j                              t          t          j        d¦  «        ¦  «        S )u]  Compute the median value of the lists in the array.

        Examples:
            >>> import duckdb
            >>> import narwhals as nw
            >>> df_native = duckdb.sql("SELECT * FROM VALUES ([1]), ([3, 4, NULL]) df(a)")
            >>> df = nw.from_native(df_native)
            >>> df.with_columns(a_median=nw.col("a").list.median())
            â”Œâ”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”
            |    Narwhals LazyFrame     |
            |---------------------------|
            |â”Œâ”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”¬â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”|
            |â”‚      a       â”‚ a_median â”‚|
            |â”‚   int32[]    â”‚  double  â”‚|
            |â”œâ”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”¼â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”¤|
            |â”‚ [1]          â”‚      1.0 â”‚|
            |â”‚ [3, 4, NULL] â”‚      3.5 â”‚|
            |â””â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”´â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”˜|
            â””â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”˜
        zlist.medianr   r   s    r   ÚmedianzExprListNamespace.medianÚ   s'   € ð* Œz×&Ò&¥xµÔ0DÀmÑ'TÔ'TÑUÔUÐUr   c                óf   — | j                              t          t          j        d¦  «        ¦  «        S )uY  Compute the sum value of the lists in the array.

        Examples:
            >>> import polars as pl
            >>> import narwhals as nw
            >>> df_native = pl.DataFrame({"a": [[1], [3, 4, None]]})
            >>> df = nw.from_native(df_native)
            >>> df.with_columns(a_sum=nw.col("a").list.sum())
            â”Œâ”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”
            |   Narwhals DataFrame   |
            |------------------------|
            |shape: (2, 2)           |
            |â”Œâ”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”¬â”€â”€â”€â”€â”€â”€â”€â”|
            |â”‚ a            â”† a_sum â”‚|
            |â”‚ ---          â”† ---   â”‚|
            |â”‚ list[i64]    â”† i64   â”‚|
            |â•žâ•â•â•â•â•â•â•â•â•â•â•â•â•â•â•ªâ•â•â•â•â•â•â•â•¡|
            |â”‚ [1]          â”† 1     â”‚|
            |â”‚ [3, 4, null] â”† 7     â”‚|
            |â””â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”´â”€â”€â”€â”€â”€â”€â”€â”˜|
            â””â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”˜
        zlist.sumr   r   s    r   ÚsumzExprListNamespace.sumñ   r1   r   F©Ú
descendingÚ
nulls_lastr9   Úboolr:   c               ól   — | j                              t          t          j        d||¬¦  «        ¦  «        S )uk  Sort the lists of the expression.

        Arguments:
            descending: Sort in descending order.
            nulls_last: Place null values last.

        Examples:
            >>> import duckdb
            >>> import narwhals as nw
            >>> df_native = duckdb.sql(
            ...     "SELECT * FROM VALUES ([2, -1, 1]), ([3, -4, NULL]) df(a)"
            ... )
            >>> df = nw.from_native(df_native)
            >>> df.with_columns(a_sorted=nw.col("a").list.sort())
            â”Œâ”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”
            |       Narwhals LazyFrame        |
            |---------------------------------|
            |â”Œâ”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”¬â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”|
            |â”‚       a       â”‚   a_sorted    â”‚|
            |â”‚    int32[]    â”‚    int32[]    â”‚|
            |â”œâ”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”¼â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”¤|
            |â”‚ [2, -1, 1]    â”‚ [-1, 1, 2]    â”‚|
            |â”‚ [3, -4, NULL] â”‚ [NULL, -4, 3] â”‚|
            |â””â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”´â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”˜|
            â””â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”˜
        z	list.sortr8   r   )r   r9   r:   s      r   ÚsortzExprListNamespace.sort
  s?   € ð6 Œz×&Ò&ÝÝÔ$ØØ%Ø%ð	ñ ô ñ
ô 
ð 	
r   N)r   r   r   r   )r   r   )r    r
   r   r   )r#   r$   r   r   )r9   r;   r:   r;   r   r   )r(   Ú
__module__Ú__qualname__r   r   r   r"   r,   r.   r0   r3   r5   r7   r=   © r   r   r   r      s  € € € € € ðð ð ð ðSð Sð Sð Sð:Vð Vð Vð Vð:
ð 
ð 
ð 
ð>&
ð &
ð &
ð &
ðPSð Sð Sð Sð.Sð Sð Sð Sð2Tð Tð Tð Tð.Vð Vð Vð Vð.Sð Sð Sð Sð2 */À5ð "
ð "
ð "
ð "
ð "
ð "
ð "
ð "
r   r   N)Ú
__future__r   Útypingr   r   r   Únarwhals._expression_parsingr   r   Únarwhals.exprr	   Únarwhals.typingr
   r   r   r@   r   r   ú<module>rF      sÏ   ðØ "Ð "Ð "Ð "Ð "Ð "à 2Ð 2Ð 2Ð 2Ð 2Ð 2Ð 2Ð 2Ð 2Ð 2à ;Ð ;Ð ;Ð ;Ð ;Ð ;Ð ;Ð ;àð 1Ø"Ð"Ð"Ð"Ð"Ð"Ø0Ð0Ð0Ð0Ð0Ð0àˆ˜vÐ&Ñ&Ô&€ð^
ð ^
ð ^
ð ^
ð ^
˜ œñ ^
ô ^
ð ^
ð ^
ð ^
r   