39 lines
1.2 KiB
Python
39 lines
1.2 KiB
Python
|
from __future__ import annotations
|
||
|
|
|
||
|
|
from typing import TYPE_CHECKING
|
||
|
|
|
||
|
|
from narwhals._compliant import CompliantSelector, EagerSelectorNamespace
|
||
|
|
from narwhals._pandas_like.expr import PandasLikeExpr
|
||
|
|
|
||
|
|
if TYPE_CHECKING:
|
||
|
|
from narwhals._compliant.typing import ScalarKwargs
|
||
|
|
from narwhals._pandas_like.dataframe import PandasLikeDataFrame # noqa: F401
|
||
|
|
from narwhals._pandas_like.series import PandasLikeSeries # noqa: F401
|
||
|
|
|
||
|
|
|
||
|
|
class PandasSelectorNamespace(
|
||
|
|
EagerSelectorNamespace["PandasLikeDataFrame", "PandasLikeSeries"]
|
||
|
|
):
|
||
|
|
@property
|
||
|
|
def _selector(self) -> type[PandasSelector]:
|
||
|
|
return PandasSelector
|
||
|
|
|
||
|
|
|
||
|
|
class PandasSelector( # type: ignore[misc]
|
||
|
|
CompliantSelector["PandasLikeDataFrame", "PandasLikeSeries"], PandasLikeExpr
|
||
|
|
):
|
||
|
|
_depth: int = 0
|
||
|
|
_scalar_kwargs: ScalarKwargs = {} # noqa: RUF012
|
||
|
|
_function_name: str = "selector"
|
||
|
|
|
||
|
|
def _to_expr(self) -> PandasLikeExpr:
|
||
|
|
return PandasLikeExpr(
|
||
|
|
self._call,
|
||
|
|
depth=self._depth,
|
||
|
|
function_name=self._function_name,
|
||
|
|
evaluate_output_names=self._evaluate_output_names,
|
||
|
|
alias_output_names=self._alias_output_names,
|
||
|
|
implementation=self._implementation,
|
||
|
|
version=self._version,
|
||
|
|
)
|