np_xdarray_intp

Author: cmp0xff

pandas-stubs/_typing.pyi

@@ -837,6 +837,7 @@ StrLike: TypeAlias = str | np.str_
 
 ScalarT = TypeVar("ScalarT", bound=Scalar)
 # Refine the definitions below in 3.9 to use the specialized type.
+np_ndarray_intp: TypeAlias = npt.NDArray[np.intp]
 np_ndarray_int64: TypeAlias = npt.NDArray[np.int64]
 np_ndarray_int: TypeAlias = npt.NDArray[np.signedinteger]
 np_ndarray_anyint: TypeAlias = npt.NDArray[np.integer]
@@ -862,6 +863,7 @@ np_ndarray: TypeAlias = np.ndarray[ShapeT, np.dtype[GenericT]]
 # Numpy arrays with known shape (Do not use as argument types, only as return types)
 np_1darray: TypeAlias = np.ndarray[tuple[int], np.dtype[GenericT]]
 np_1darray_bool: TypeAlias = np_1darray[np.bool]
+np_1darray_intp: TypeAlias = np_1darray[np.intp]
 np_1darray_int64: TypeAlias = np_1darray[np.int64]
 np_1darray_float: TypeAlias = np_1darray[np.floating]
 

pandas-stubs/core/arrays/base.pyi

@@ -22,6 +22,7 @@ from pandas._typing import (
     SequenceIndexer,
     TakeIndexer,
     np_1darray,
+    np_1darray_intp,
     npt,
 )
 
@@ -75,7 +76,7 @@ class ExtensionArray:
         value: ListLike,
         side: Literal["left", "right"] = ...,
         sorter: ListLike | None = ...,
-    ) -> np_1darray[np.intp]: ...
+    ) -> np_1darray_intp: ...
     @overload
     def searchsorted(
         self,

pandas-stubs/core/base.pyi

@@ -44,6 +44,7 @@ from pandas._typing import (
     Scalar,
     SupportsDType,
     np_1darray,
+    np_1darray_intp,
     np_ndarray_anyint,
     np_ndarray_bool,
     np_ndarray_complex,
@@ -163,7 +164,7 @@ class IndexOpsMixin(OpsMixin, Generic[S1, GenericT_co]):
         value: ListLike,
         side: Literal["left", "right"] = ...,
         sorter: ListLike | None = None,
-    ) -> np_1darray[np.intp]: ...
+    ) -> np_1darray_intp: ...
     @overload
     def searchsorted(
         self,

pandas-stubs/core/indexers.pyi

@@ -5,6 +5,8 @@ from numpy import typing as npt
 from pandas._typing import (
     AnyArrayLike,
     np_1darray_bool,
+    np_1darray_intp,
+    np_ndarray_intp,
 )
 
 def check_array_indexer(
@@ -14,7 +16,7 @@ def check_array_indexer(
 class BaseIndexer:
     def __init__(
         self,
-        index_array: npt.NDArray[Any] | None = ...,
+        index_array: np_ndarray_intp | None = ...,
         window_size: int = ...,
         **kwargs: Any,
     ) -> None: ...
@@ -24,12 +26,12 @@ class BaseIndexer:
         min_periods: int | None = ...,
         center: bool | None = ...,
         closed: str | None = ...,
-    ) -> tuple[npt.NDArray[Any], npt.NDArray[Any]]: ...
+    ) -> tuple[np_1darray_intp, np_1darray_intp]: ...
 
 class VariableOffsetWindowIndexer(BaseIndexer):
     def __init__(
         self,
-        index_array: npt.NDArray[Any] | None = ...,
+        index_array: np_ndarray_intp | None = ...,
         window_size: int = ...,
         index=...,
         offset=...,

pandas-stubs/core/indexers/objects.pyi

@@ -4,11 +4,12 @@ from numpy import typing as npt
 from pandas.core.indexes.datetimes import DatetimeIndex
 
 from pandas._libs.tslibs import BaseOffset
+from pandas._typing import np_ndarray_intp
 
 class BaseIndexer:
     def __init__(
         self,
-        index_array: npt.NDArray[Any] | None = None,
+        index_array: np_ndarray_intp | None = None,
         window_size: int = 0,
         **kwargs: Any,
     ) -> None: ...
@@ -26,7 +27,7 @@ class FixedForwardWindowIndexer(BaseIndexer): ...
 class VariableOffsetWindowIndexer(BaseIndexer):
     def __init__(
         self,
-        index_array: npt.NDArray[Any] | None = None,
+        index_array: np_ndarray_intp | None = None,
         window_size: int = 0,
         index: DatetimeIndex | None = None,
         offset: BaseOffset | None = None,

pandas-stubs/core/indexes/base.pyi

@@ -107,6 +107,7 @@ from pandas._typing import (
     TimedeltaDtypeArg,
     TimestampDtypeArg,
     np_1darray,
+    np_1darray_intp,
     np_ndarray,
     np_ndarray_anyint,
     np_ndarray_bool,
@@ -453,15 +454,15 @@ class Index(IndexOpsMixin[S1], ElementOpsMixin[S1]):
         method: ReindexMethod | None = None,
         limit: int | None = None,
         tolerance: Scalar | AnyArrayLike | Sequence[Scalar] | None = None,
-    ) -> np_1darray[np.intp]: ...
+    ) -> np_1darray_intp: ...
     def reindex(
         self,
         target: Iterable[Any],
         method: ReindexMethod | None = None,
         level: int | None = None,
         limit: int | None = None,
         tolerance: Scalar | AnyArrayLike | Sequence[Scalar] | None = None,
-    ) -> tuple[Index, np_1darray[np.intp] | None]: ...
+    ) -> tuple[Index, np_1darray_intp | None]: ...
     @overload
     def join(
         self,
@@ -471,7 +472,7 @@ class Index(IndexOpsMixin[S1], ElementOpsMixin[S1]):
         level: Level | None = None,
         return_indexers: Literal[True],
         sort: bool = False,
-    ) -> tuple[Index, np_1darray[np.intp] | None, np_1darray[np.intp] | None]: ...
+    ) -> tuple[Index, np_1darray_intp | None, np_1darray_intp | None]: ...
     @overload
     def join(
         self,
@@ -530,7 +531,7 @@ class Index(IndexOpsMixin[S1], ElementOpsMixin[S1]):
     ): ...
     @final
     def sort(self, *args: Any, **kwargs: Any) -> None: ...
-    def argsort(self, *args: Any, **kwargs: Any) -> np_1darray[np.intp]: ...
+    def argsort(self, *args: Any, **kwargs: Any) -> np_1darray_intp: ...
     def get_indexer_non_unique(self, target): ...
     @final
     def get_indexer_for(self, target, **kwargs: Any): ...

pandas-stubs/core/indexes/category.pyi

@@ -3,7 +3,6 @@ from collections.abc import (
     Iterable,
 )
 
-import numpy as np
 from pandas.core.accessor import PandasDelegate
 from pandas.core.arrays.categorical import Categorical
 from pandas.core.indexes.base import Index
@@ -14,11 +13,11 @@ from pandas._typing import (
     S1,
     Dtype,
     ListLike,
-    np_1darray,
+    np_1darray_intp,
 )
 
 class CategoricalIndex(ExtensionIndex[S1], PandasDelegate):
-    codes: np_1darray[np.intp] = ...
+    codes: np_1darray_intp = ...
     categories: Index[S1] = ...
     @property
     def array(self) -> Categorical: ...  # type: ignore[override] # pyrefly: ignore[bad-override]

pandas-stubs/core/indexes/datetimes.pyi

@@ -34,7 +34,7 @@ from pandas._typing import (
     IntervalClosedType,
     TimeUnit,
     TimeZones,
-    np_1darray,
+    np_1darray_intp,
     np_ndarray,
     np_ndarray_dt,
     np_ndarray_td,
@@ -91,14 +91,14 @@ class DatetimeIndex(
     def inferred_type(self) -> str: ...
     def indexer_at_time(
         self, time: str | time, asof: bool = False
-    ) -> np_1darray[np.intp]: ...
+    ) -> np_1darray_intp: ...
     def indexer_between_time(
         self,
         start_time: time | str,
         end_time: time | str,
         include_start: bool = True,
         include_end: bool = True,
-    ) -> np_1darray[np.intp]: ...
+    ) -> np_1darray_intp: ...
     def to_julian_date(self) -> Index[float]: ...
     def isocalendar(self) -> DataFrame: ...
     @property

pandas-stubs/core/indexes/period.pyi

@@ -23,6 +23,7 @@ from pandas._typing import (
     Dtype,
     Frequency,
     np_1darray,
+    np_1darray_intp,
     np_ndarray_bool,
 )
 
@@ -67,7 +68,7 @@ class PeriodIndex(DatetimeIndexOpsMixin[pd.Period, np.object_], PeriodIndexField
     ) -> NaTType: ...
     def asof_locs(
         self, where: pd.DatetimeIndex | Self, mask: np_ndarray_bool
-    ) -> np_1darray[np.intp]: ...
+    ) -> np_1darray_intp: ...
     @property
     def is_full(self) -> bool: ...
     @property

pandas-stubs/core/indexes/range.pyi

@@ -22,7 +22,7 @@ from pandas._typing import (
     HashableT,
     MaskType,
     Scalar,
-    np_1darray,
+    np_1darray_intp,
     np_ndarray_anyint,
     np_ndarray_bool,
 )
@@ -62,7 +62,7 @@ class RangeIndex(_IndexSubclassBase[int, np.int64]):
     def has_duplicates(self) -> bool: ...
     def factorize(
         self, sort: bool = False, use_na_sentinel: bool = True
-    ) -> tuple[np_1darray[np.intp], RangeIndex]: ...
+    ) -> tuple[np_1darray_intp, RangeIndex]: ...
     @property
     def size(self) -> int: ...
     def all(self, *args: Any, **kwargs: Any) -> bool: ...