comments

Author: cmp0xff

pandas-stubs/core/frame.pyi

@@ -164,6 +164,7 @@ from pandas._typing import (
     WriteBuffer,
     XMLParsers,
     np_2darray,
+    np_ndarray_bool,
     np_ndarray_float,
 )
 
@@ -2200,7 +2201,7 @@ class DataFrame(NDFrame, OpsMixin, _GetItemHack):
         cond: (
             Series
             | DataFrame
-            | npt.NDArray[Any]
+            | np_ndarray_bool
             | Callable[[DataFrame], DataFrame]
             | Callable[[Any], _bool]
         ),
@@ -2216,7 +2217,7 @@ class DataFrame(NDFrame, OpsMixin, _GetItemHack):
         cond: (
             Series
             | DataFrame
-            | npt.NDArray[Any]
+            | np_ndarray_bool
             | Callable[[DataFrame], DataFrame]
             | Callable[[Any], _bool]
         ),
@@ -2643,7 +2644,7 @@ class DataFrame(NDFrame, OpsMixin, _GetItemHack):
         cond: (
             Series
             | DataFrame
-            | npt.NDArray[Any]
+            | np_ndarray_bool
             | Callable[[DataFrame], DataFrame]
             | Callable[[Any], _bool]
         ),
@@ -2659,7 +2660,7 @@ class DataFrame(NDFrame, OpsMixin, _GetItemHack):
         cond: (
             Series
             | DataFrame
-            | npt.NDArray[Any]
+            | np_ndarray_bool
             | Callable[[DataFrame], DataFrame]
             | Callable[[Any], _bool]
         ),

pandas-stubs/core/groupby/groupby.pyi

@@ -64,8 +64,8 @@ from pandas._typing import (
     TimestampConvertibleTypes,
     WindowingEngine,
     WindowingEngineKwargs,
+    np_ndarray_dt,
     np_ndarray_int64,
-    npt,
 )
 
 from pandas.plotting import PlotAccessor
@@ -241,7 +241,7 @@ class GroupBy(BaseGroupBy[NDFrameT]):
         adjust: bool = ...,
         ignore_na: bool = ...,
         axis: Axis = ...,
-        times: str | npt.NDArray[Any] | Series | np.timedelta64 | None = ...,
+        times: str | np_ndarray_dt | Series | np.timedelta64 | None = ...,
         method: CalculationMethod = ...,
         *,
         selection: IndexLabel | None = ...,

pandas-stubs/core/series.pyi

@@ -1398,7 +1398,7 @@ class Series(IndexOpsMixin[S1], ElementOpsMixin[S1], NDFrame):
         n: int | None = None,
         frac: float | None = None,
         replace: _bool = False,
-        weights: _str | ListLike | npt.NDArray[Any] | None = None,
+        weights: _str | ListLike | np_ndarray_float | None = None,
         random_state: RandomState | None = None,
         axis: AxisIndex | None = None,
         ignore_index: _bool = False,
@@ -1613,7 +1613,7 @@ class Series(IndexOpsMixin[S1], ElementOpsMixin[S1], NDFrame):
         cond: (
             Series[S1]
             | Series[_bool]
-            | npt.NDArray[Any]
+            | np_ndarray_bool
             | Callable[[Series[S1]], Series[bool]]
             | Callable[[S1], bool]
         ),
@@ -1629,7 +1629,7 @@ class Series(IndexOpsMixin[S1], ElementOpsMixin[S1], NDFrame):
         cond: (
             Series[S1]
             | Series[_bool]
-            | npt.NDArray[Any]
+            | np_ndarray_bool
             | Callable[[Series[S1]], Series[bool]]
             | Callable[[S1], bool]
         ),
@@ -1645,7 +1645,7 @@ class Series(IndexOpsMixin[S1], ElementOpsMixin[S1], NDFrame):
         cond: (
             Series[S1]
             | Series[_bool]
-            | npt.NDArray[Any]
+            | np_ndarray_bool
             | Callable[[Series[S1]], Series[bool]]
             | Callable[[S1], bool]
         ),
@@ -1661,7 +1661,7 @@ class Series(IndexOpsMixin[S1], ElementOpsMixin[S1], NDFrame):
         cond: (
             Series[S1]
             | Series[_bool]
-            | npt.NDArray[Any]
+            | np_ndarray_bool
             | Callable[[Series[S1]], Series[bool]]
             | Callable[[S1], bool]
         ),
@@ -4127,7 +4127,7 @@ class Series(IndexOpsMixin[S1], ElementOpsMixin[S1], NDFrame):
         adjust: _bool = True,
         ignore_na: _bool = False,
         axis: Axis = 0,
-        times: npt.NDArray[Any] | Series | None = None,
+        times: np_ndarray_dt | Series | None = None,
         method: CalculationMethod = "single",
     ) -> ExponentialMovingWindow[Series]: ...
     @final