Skip to content

Commit d8cc346

Browse files
author
OlivierLuG
committed
added a test for issue pandas-dev#20627
1 parent 6258397 commit d8cc346

File tree

1 file changed

+20
-0
lines changed

1 file changed

+20
-0
lines changed

pandas/tests/frame/test_nonunique_indexes.py

+20
Original file line numberDiff line numberDiff line change
@@ -513,3 +513,23 @@ def test_set_value_by_index(self):
513513

514514
df.iloc[:, 0] = 3
515515
tm.assert_series_equal(df.iloc[:, 1], expected)
516+
517+
518+
def test_iloc_with_boolean_operation():
519+
# GH 20627
520+
result = DataFrame([[0, 1], [2, 3], [4, 5], [6, np.nan]])
521+
result.iloc[result.index <= 2] *= 2
522+
expected = DataFrame([[0, 2], [4, 6], [8, 10], [6, np.nan]])
523+
tm.assert_frame_equal(result, expected)
524+
525+
result.iloc[result.index > 2] *= 2
526+
expected = DataFrame([[0, 2], [4, 6], [8, 10], [12, np.nan]])
527+
tm.assert_frame_equal(result, expected)
528+
529+
result.iloc[[True, True, False, False]] *= 2
530+
expected = DataFrame([[0, 4], [8, 12], [8, 10], [12, np.nan]])
531+
tm.assert_frame_equal(result, expected)
532+
533+
result.iloc[[False, False, True, True]] /= 2
534+
expected = DataFrame([[0.0, 4.0], [8.0, 12.0], [4.0, 5.0], [6.0, np.nan]])
535+
tm.assert_frame_equal(result, expected)

0 commit comments

Comments
 (0)