GH1255 Initial commit

Author: loicdiridollou

pandas-stubs/_typing.pyi

@@ -14,6 +14,7 @@ from re import Pattern
 import sys
 from typing import (
     Any,
+    KeysView,
     Literal,
     Protocol,
     SupportsIndex,
@@ -796,9 +797,6 @@ SliceType: TypeAlias = Hashable | None
 
 num: TypeAlias = complex
 
-# AxesData is used for data for Index
-AxesData: TypeAlias = Axes | dict
-
 DtypeNp = TypeVar("DtypeNp", bound=np.dtype[np.generic])
 KeysArgType: TypeAlias = Any
 ListLikeT = TypeVar("ListLikeT", bound=ListLike)
@@ -853,6 +851,9 @@ IndexingInt: TypeAlias = (
     int | np.int_ | np.integer | np.unsignedinteger | np.signedinteger | np.int8
 )
 
+# AxesData is used for data for Index
+AxesData: TypeAlias = Mapping[SeriesDType, Any] | Axes | KeysView
+
 # Any plain Python or numpy function
 Function: TypeAlias = np.ufunc | Callable[..., Any]
 # Use a distinct HashableT in shared types to avoid conflicts with

pandas-stubs/core/series.pyi

@@ -111,6 +111,7 @@ from pandas._typing import (
     AnyArrayLike,
     ArrayLike,
     Axes,
+    AxesData,
     Axis,
     AxisColumn,
     AxisIndex,
@@ -252,7 +253,7 @@ class Series(IndexOpsMixin[S1], NDFrame):
     def __new__(
         cls,
         data: npt.NDArray[np.float64],
-        index: Axes | None = ...,
+        index: AxesData | None = ...,
         dtype: Dtype = ...,
         name: Hashable = ...,
         copy: bool = ...,
@@ -261,7 +262,7 @@ class Series(IndexOpsMixin[S1], NDFrame):
     def __new__(  # type: ignore[overload-overlap] # pyright: ignore[reportOverlappingOverload]
         cls,
         data: Sequence[Never],
-        index: Axes | None = ...,
+        index: AxesData | None = ...,
         dtype: Dtype = ...,
         name: Hashable = ...,
         copy: bool = ...,
@@ -270,7 +271,7 @@ class Series(IndexOpsMixin[S1], NDFrame):
     def __new__(
         cls,
         data: Sequence[list[_str]],
-        index: Axes | None = ...,
+        index: AxesData | None = ...,
         dtype: Dtype = ...,
         name: Hashable = ...,
         copy: bool = ...,
@@ -279,7 +280,7 @@ class Series(IndexOpsMixin[S1], NDFrame):
     def __new__(
         cls,
         data: Sequence[_str],
-        index: Axes | None = ...,
+        index: AxesData | None = ...,
         dtype: Dtype = ...,
         name: Hashable = ...,
         copy: bool = ...,
@@ -295,7 +296,7 @@ class Series(IndexOpsMixin[S1], NDFrame):
             | datetime
             | date
         ),
-        index: Axes | None = ...,
+        index: AxesData | None = ...,
         dtype: TimestampDtypeArg = ...,
         name: Hashable = ...,
         copy: bool = ...,
@@ -304,7 +305,7 @@ class Series(IndexOpsMixin[S1], NDFrame):
     def __new__(
         cls,
         data: _ListLike,
-        index: Axes | None = ...,
+        index: AxesData | None = ...,
         *,
         dtype: TimestampDtypeArg,
         name: Hashable = ...,
@@ -314,7 +315,7 @@ class Series(IndexOpsMixin[S1], NDFrame):
     def __new__(
         cls,
         data: PeriodIndex | Sequence[Period],
-        index: Axes | None = ...,
+        index: AxesData | None = ...,
         dtype: PeriodDtype = ...,
         name: Hashable = ...,
         copy: bool = ...,
@@ -329,7 +330,7 @@ class Series(IndexOpsMixin[S1], NDFrame):
             | np.timedelta64
             | timedelta
         ),
-        index: Axes | None = ...,
+        index: AxesData | None = ...,
         dtype: TimedeltaDtypeArg = ...,
         name: Hashable = ...,
         copy: bool = ...,
@@ -343,7 +344,7 @@ class Series(IndexOpsMixin[S1], NDFrame):
             | Sequence[Interval[_OrderableT]]
             | dict[HashableT1, Interval[_OrderableT]]
         ),
-        index: Axes | None = ...,
+        index: AxesData | None = ...,
         dtype: Literal["Interval"] = ...,
         name: Hashable = ...,
         copy: bool = ...,
@@ -352,7 +353,7 @@ class Series(IndexOpsMixin[S1], NDFrame):
     def __new__(  # type: ignore[overload-overlap]
         cls,
         data: Scalar | _ListLike | dict[HashableT1, Any] | None,
-        index: Axes | None = ...,
+        index: AxesData | None = ...,
         *,
         dtype: type[S1],
         name: Hashable = ...,
@@ -362,7 +363,7 @@ class Series(IndexOpsMixin[S1], NDFrame):
     def __new__(  # type: ignore[overload-overlap] # pyright: ignore[reportOverlappingOverload]
         cls,
         data: Sequence[bool],
-        index: Axes | None = ...,
+        index: AxesData | None = ...,
         dtype: Dtype = ...,
         name: Hashable = ...,
         copy: bool = ...,
@@ -371,7 +372,7 @@ class Series(IndexOpsMixin[S1], NDFrame):
     def __new__(  # type: ignore[overload-overlap]
         cls,
         data: Sequence[int],
-        index: Axes | None = ...,
+        index: AxesData | None = ...,
         dtype: Dtype = ...,
         name: Hashable = ...,
         copy: bool = ...,
@@ -380,7 +381,7 @@ class Series(IndexOpsMixin[S1], NDFrame):
     def __new__(
         cls,
         data: Sequence[float],
-        index: Axes | None = ...,
+        index: AxesData | None = ...,
         dtype: Dtype = ...,
         name: Hashable = ...,
         copy: bool = ...,
@@ -389,7 +390,7 @@ class Series(IndexOpsMixin[S1], NDFrame):
     def __new__(  # type: ignore[overload-cannot-match] # pyright: ignore[reportOverlappingOverload]
         cls,
         data: Sequence[int | float],
-        index: Axes | None = ...,
+        index: AxesData | None = ...,
         dtype: Dtype = ...,
         name: Hashable = ...,
         copy: bool = ...,
@@ -398,7 +399,7 @@ class Series(IndexOpsMixin[S1], NDFrame):
     def __new__(
         cls,
         data: S1 | _ListLike[S1] | dict[HashableT1, S1] | dict_keys[S1, Any],
-        index: Axes | None = ...,
+        index: AxesData | None = ...,
         dtype: Dtype = ...,
         name: Hashable = ...,
         copy: bool = ...,
@@ -415,7 +416,7 @@ class Series(IndexOpsMixin[S1], NDFrame):
             | NAType
             | None
         ) = ...,
-        index: Axes | None = ...,
+        index: AxesData | None = ...,
         dtype: Dtype = ...,
         name: Hashable = ...,
         copy: bool = ...,

tests/test_series.py

@@ -4,6 +4,7 @@
     Hashable,
     Iterable,
     Iterator,
+    Mapping,
     Sequence,
 )
 import datetime
@@ -44,6 +45,7 @@
 import xarray as xr
 
 from pandas._typing import (
+    AxesData,
     DtypeObj,
     Scalar,
 )
@@ -867,7 +869,6 @@ def test_types_scalar_arithmetic() -> None:
     res_pow3: pd.Series = s.pow(0.5)
 
 
-def test_types_complex_arithmetic() -> None:
     # GH 103
     c = 1 + 1j
     s = pd.Series([1.0, 2.0, 3.0])
@@ -3922,3 +3923,15 @@ def test_series_unstack() -> None:
         ),
         pd.DataFrame,
     )
+
+
+def test_series_index_type() -> None:
+    index = {"a":3, "c": 4}
+    lst = [1, 2]
+    assert_type(index, Mapping[str, Any])
+
+    check(assert_type(pd.Series(lst, index=index), "pd.Series[int]"), pd.Series, np.integer)
+    check(assert_type(pd.Series([1, 2], index=index.keys()), "pd.Series[int]"), pd.Series, np.integer)
+
+    if TYPE_CHECKING_INVALID_USAGE:
+        t = pd.Series([1,2], index="ab")  # type: ignore # pyright: ignore[reportCallIssue, reportArgumentType]