Skip to content

Commit c524947

Browse files
committed
partial list-like indexing
1 parent 464cff2 commit c524947

File tree

2 files changed

+6
-10
lines changed

2 files changed

+6
-10
lines changed

pandas/indexes/interval.py

+2-5
Original file line numberDiff line numberDiff line change
@@ -412,11 +412,8 @@ def _convert_list_indexer(self, keyarr, kind=None):
412412
included in the categories
413413
"""
414414
locs = self.get_indexer(keyarr)
415-
# TODO: handle keyarr if it includes intervals
416-
if (locs == -1).any():
417-
raise KeyError("a list-indexer must only include "
418-
"existing intervals")
419-
415+
check = locs == -1
416+
locs = locs[~check]
420417
return locs
421418

422419
def _check_method(self, method):

pandas/tests/indexing/test_interval.py

+4-5
Original file line numberDiff line numberDiff line change
@@ -61,8 +61,7 @@ def f():
6161

6262
self.assertRaises(KeyError, f)
6363

64-
# this is a bit different than other .loc indexing
65-
def f():
66-
df.loc[[10, 4]]
67-
68-
self.assertRaises(KeyError, f)
64+
# partial missing
65+
result = df.loc[[10, 4]]
66+
expected = df.iloc[4:6]
67+
tm.assert_frame_equal(result, expected)

0 commit comments

Comments
 (0)