blob: cb976d4e8f5997748667cb1a029c7eec40bdd091 (
plain)
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
|
from __future__ import annotations
from typing import Generic
from narwhals.typing import SeriesT
class SeriesCatNamespace(Generic[SeriesT]):
def __init__(self, series: SeriesT) -> None:
self._narwhals_series = series
def get_categories(self) -> SeriesT:
"""Get unique categories from column.
Returns:
A new Series containing the unique categories.
Examples:
>>> import pandas as pd
>>> import narwhals as nw
>>> s_native = pd.Series(["apple", "mango", "mango"], dtype="category")
>>> s = nw.from_native(s_native, series_only=True)
>>> s.cat.get_categories().to_native()
0 apple
1 mango
dtype: object
"""
return self._narwhals_series._with_compliant(
self._narwhals_series._compliant_series.cat.get_categories()
)
|
