Skip to content

Commit 68780c7

Browse files
authored
Add tuple as possible arguement for axes (#447)
1 parent dc5ea0e commit 68780c7

File tree

2 files changed

+9
-1
lines changed

2 files changed

+9
-1
lines changed

pandas-stubs/_typing.pyi

+1-1
Original file line numberDiff line numberDiff line change
@@ -120,7 +120,7 @@ Level: TypeAlias = Union[Hashable, int]
120120
Suffixes: TypeAlias = tuple[Optional[str], Optional[str]]
121121
Ordered: TypeAlias = Optional[bool]
122122
JSONSerializable: TypeAlias = Union[PythonScalar, list, dict]
123-
Axes: TypeAlias = Union[AnyArrayLike, list, dict, range]
123+
Axes: TypeAlias = Union[AnyArrayLike, list, dict, range, tuple]
124124
Renamer: TypeAlias = Union[Mapping[Any, Label], Callable[[Any], Label]]
125125
T = TypeVar("T")
126126
FuncType: TypeAlias = Callable[..., Any]

tests/test_frame.py

+8
Original file line numberDiff line numberDiff line change
@@ -2044,6 +2044,14 @@ def test_series_groupby_and_value_counts() -> None:
20442044
check(assert_type(c, pd.Series), pd.Series)
20452045

20462046

2047+
def test_axes_as_tuple() -> None:
2048+
# GH 384
2049+
index = (3, 5, 7)
2050+
columns = ["a", "b", "c"]
2051+
df = pd.DataFrame(data=1, index=index, columns=columns)
2052+
check(assert_type(df, pd.DataFrame), pd.DataFrame)
2053+
2054+
20472055
def test_setitem_none() -> None:
20482056
df = pd.DataFrame(
20492057
{"A": [1, 2, 3], "B": ["abc", "def", "ghi"]}, index=["x", "y", "z"]

0 commit comments

Comments
 (0)