blob: 912da70dd0c471f5022b86f92cf1b9c94ae36e01 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
|
from __future__ import annotations
from typing import TYPE_CHECKING
from narwhals._compliant.any_namespace import CatNamespace
from narwhals._pandas_like.utils import PandasLikeSeriesNamespace
if TYPE_CHECKING:
from narwhals._pandas_like.series import PandasLikeSeries
class PandasLikeSeriesCatNamespace(
PandasLikeSeriesNamespace, CatNamespace["PandasLikeSeries"]
):
def get_categories(self) -> PandasLikeSeries:
s = self.native
return self.with_native(type(s)(s.cat.categories, name=s.name))
|
