Skip to content

Commit ef50ef8

Browse files
GH677 Allow pd.RangeIndex to be initialized with range
1 parent a4615d5 commit ef50ef8

File tree

2 files changed

+8
-2
lines changed

2 files changed

+8
-2
lines changed

pandas-stubs/core/indexes/range.pyi

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@ from pandas._typing import (
1515
class RangeIndex(Index[int]):
1616
def __new__(
1717
cls,
18-
start: int | RangeIndex = ...,
18+
start: int | RangeIndex | range = ...,
1919
stop: int = ...,
2020
step: int = ...,
2121
dtype=...,
@@ -24,7 +24,7 @@ class RangeIndex(Index[int]):
2424
): ...
2525
def __init__(
2626
self,
27-
start: int | RangeIndex = ...,
27+
start: int | RangeIndex | range = ...,
2828
stop: int = ...,
2929
step: int = ...,
3030
dtype=...,

tests/test_indexes.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -912,6 +912,12 @@ def test_getitem() -> None:
912912
check(assert_type(i0[[0, 2]], "pd.Index[str]"), pd.Index, str)
913913

914914

915+
def test_range_index_range() -> None:
916+
"""Test that pd.RangeIndex can be initialized from range."""
917+
iri = pd.RangeIndex(range(5))
918+
check(assert_type(iri, pd.RangeIndex), pd.RangeIndex, int)
919+
920+
915921
def test_multiindex_dtypes():
916922
# GH-597
917923
mi = pd.MultiIndex.from_tuples([(1, 2.0), (2, 3.0)], names=["foo", "bar"])

0 commit comments

Comments
 (0)