Skip to content

Commit 14a581d

Browse files
authored
REGR: isin raising TypeError for RangeIndex (#41155)
1 parent 2853622 commit 14a581d

File tree

2 files changed

+9
-1
lines changed

2 files changed

+9
-1
lines changed

pandas/core/algorithms.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -466,7 +466,7 @@ def isin(comps: AnyArrayLike, values: AnyArrayLike) -> np.ndarray:
466466
# Avoid raising in extract_array
467467
values = np.array(values)
468468
else:
469-
values = extract_array(values, extract_numpy=True)
469+
values = extract_array(values, extract_numpy=True, extract_range=True)
470470

471471
comps = _ensure_arraylike(comps)
472472
comps = extract_array(comps, extract_numpy=True)

pandas/tests/indexes/ranges/test_range.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -524,3 +524,11 @@ def test_append_len_one(self, RI):
524524
# GH39401
525525
result = RI.append([])
526526
tm.assert_index_equal(result, RI, exact=True)
527+
528+
@pytest.mark.parametrize("base", [RangeIndex(0, 2), Index([0, 1])])
529+
def test_isin_range(self, base):
530+
# GH#41151
531+
values = RangeIndex(0, 1)
532+
result = base.isin(values)
533+
expected = np.array([True, False])
534+
tm.assert_numpy_array_equal(result, expected)

0 commit comments

Comments
 (0)