}ifddlmZddlmZmZddlmZerddlmZGddeeZdS)) annotations) TYPE_CHECKINGGeneric)SeriesT)NonNestedLiteralcneZdZddZddZddZdd Zdd ZddZddZ ddZ ddZ ddZ dddddZ dS)SeriesListNamespaceseriesrreturnNonec||_dS)N)_narwhals_series)selfr s H/home/jrussi/.local/lib/python3.11/site-packages/narwhals/series_list.py__init__zSeriesListNamespace.__init__ s &cx|j|jjjS)aMReturn the number of elements in each list. Null values count towards the total. Examples: >>> import pyarrow as pa >>> import narwhals as nw >>> s_native = pa.chunked_array([[[1, 2], [3, 4, None], None, []]]) >>> s = nw.from_native(s_native, series_only=True) >>> s.list.len().to_native() # doctest: +ELLIPSIS [ [ 2, 3, null, 0 ] ] )r_with_compliant_compliant_serieslistlenrs rrzSeriesListNamespace.lens7*$44  ! 3 8 < < > >   rcx|j|jjjS)asGet 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 >>> s_native = pl.Series([[1, 1, 2], [3, 3, None], None, []]) >>> s = nw.from_native(s_native, series_only=True) >>> s.list.unique().to_native() # doctest: +NORMALIZE_WHITESPACE shape: (4,) Series: '' [list[i64]] [ [1, 2] [null, 3] null [] ] )rrrruniquers rrzSeriesListNamespace.unique(s7($44  ! 3 8 ? ? A A   ritemrcz|j|jjj|S)aFCheck if sublists contain the given item. Arguments: item: Item that will be checked for membership. Examples: >>> import polars as pl >>> import narwhals as nw >>> s_native = pl.Series([[1, 2], None, []]) >>> s = nw.from_native(s_native, series_only=True) >>> s.list.contains(1).to_native() # doctest: +NORMALIZE_WHITESPACE shape: (3,) Series: '' [bool] [ true null false ] )rrrrcontains)rrs rrzSeriesListNamespace.contains@s9($44  ! 3 8 A A$ G G   rindexintc(t|ts'dt|jd}t ||dkrd|d}t ||j|jjj |S)aReturn the value by index in each list. Negative indices are not accepted. Examples: >>> import polars as pl >>> import narwhals as nw >>> s_native = pl.Series([[1, 2], [3, 4, None], [None, 5]]) >>> s = nw.from_native(s_native, series_only=True) >>> s.list.get(1).to_native() # doctest: +NORMALIZE_WHITESPACE shape: (3,) Series: '' [i64] [ 2 4 5 ] z'Index must be of type 'int'. Got type 'z ' instead.rzIndex z8 is out of bounds: should be greater than or equal to 0.) isinstancertype__name__ TypeError ValueErrorrrrrget)rrmsgs rr&zSeriesListNamespace.getXs&%%% !Z$u++:NZZZ C.. 199Z5ZZZCS// !$44  ! 3 8 < >> import polars as pl >>> import narwhals as nw >>> s_native = pl.Series([[1], [3, 4, None]]) >>> s = nw.from_native(s_native, series_only=True) >>> s.list.min().to_native() # doctest: +NORMALIZE_WHITESPACE shape: (2,) Series: '' [i64] [ 1 3 ] )rrrrminrs rr)zSeriesListNamespace.miny7 $44  ! 3 8 < < > >   rcx|j|jjjS)aCompute the max value of the lists in the array. Examples: >>> import pyarrow as pa >>> import narwhals as nw >>> s_native = pa.chunked_array([[[1], [3, 4, None]]]) >>> s = nw.from_native(s_native, series_only=True) >>> s.list.max().to_native() # doctest: +ELLIPSIS [ [ 1, 4 ] ] )rrrrmaxrs rr,zSeriesListNamespace.maxs7"$44  ! 3 8 < < > >   rcx|j|jjjS)aCompute the mean value of the lists in the array. Examples: >>> import polars as pl >>> import narwhals as nw >>> s_native = pl.Series([[1], [3, 4, None]]) >>> s = nw.from_native(s_native, series_only=True) >>> s.list.mean().to_native() # doctest: +NORMALIZE_WHITESPACE shape: (2,) Series: '' [f64] [ 1.0 3.5 ] )rrrrmeanrs rr.zSeriesListNamespace.means7 $44  ! 3 8 = = ? ?   rcx|j|jjjS)aCompute the median value of the lists in the array. Examples: >>> import pyarrow as pa >>> import narwhals as nw >>> s_native = pa.chunked_array([[[1], [3, 4, None]]]) >>> s = nw.from_native(s_native, series_only=True) >>> s.list.median().to_native() # doctest: +ELLIPSIS [ [ 1, 3 ] ] )rrrrmedianrs rr0zSeriesListNamespace.medians7"$44  ! 3 8 ? ? A A   rcx|j|jjjS)aCompute the sum value of the lists in the array. Examples: >>> import polars as pl >>> import narwhals as nw >>> s_native = pl.Series([[1], [3, 4, None]]) >>> s = nw.from_native(s_native, series_only=True) >>> s.list.sum().to_native() # doctest: +NORMALIZE_WHITESPACE shape: (2,) Series: '' [i64] [ 1 7 ] )rrrrsumrs rr2zSeriesListNamespace.sumr*rF descending nulls_lastr4boolr5c~|j|jjj||S)a^Sort the lists of the series. Arguments: descending: Sort in descending order. nulls_last: Place null values last. Examples: >>> import polars as pl >>> import narwhals as nw >>> s_native = pl.Series([[2, -1, 1], [3, -4, None]]) >>> s = nw.from_native(s_native, series_only=True) >>> s.list.sort().to_native() # doctest: +NORMALIZE_WHITESPACE shape: (2,) Series: '' [list[i64]] [ [-1, 1, 2] [null, -4, 3] ] r3)rrrrsort)rr4r5s rr8zSeriesListNamespace.sortsE($44  ! 3 8 = =%* >     rN)r rr r )r r)rrr r)rrr r)r4r6r5r6r r)r# __module__ __qualname__rrrrr&r)r,r.r0r2r8rrr r s''''    2    0    0    B    (    *    (    *    (*/5        rr N) __future__rtypingrrnarwhals.typingrrr r;rrr?s""""""))))))))######1000000l l l l l ''*l l l l l r