Skip to content

Commit a4e4909

Browse files
mroeschkejowens
authored andcommitted
TST: Partial Boolean DataFrame Indexing (pandas-dev#17186)
Closes pandas-devgh-17170
1 parent 0bfad7c commit a4e4909

File tree

1 file changed

+14
-0
lines changed

1 file changed

+14
-0
lines changed

pandas/tests/indexing/test_indexing.py

+14
Original file line numberDiff line numberDiff line change
@@ -861,6 +861,20 @@ def test_maybe_numeric_slice(self):
861861
expected = [1]
862862
assert result == expected
863863

864+
def test_partial_boolean_frame_indexing(self):
865+
# GH 17170
866+
df = pd.DataFrame(np.arange(9.).reshape(3, 3),
867+
index=list('abc'),
868+
columns=list('ABC'))
869+
index_df = pd.DataFrame(1, index=list('ab'), columns=list('AB'))
870+
result = df[index_df.notnull()]
871+
expected = pd.DataFrame(np.array([[0., 1., np.nan],
872+
[3., 4., np.nan],
873+
[np.nan] * 3]),
874+
index=list('abc'),
875+
columns=list('ABC'))
876+
tm.assert_frame_equal(result, expected)
877+
864878

865879
class TestSeriesNoneCoercion(object):
866880
EXPECTED_RESULTS = [

0 commit comments

Comments
 (0)