BUG: wrap all supported inplace methods to avoid making a copy (#12962)

Author: dkamm

pandas/core/ops.py

@@ -186,8 +186,10 @@ def add_special_arithmetic_methods(cls, arith_method=None,
     arith_method : function (optional)
         factory for special arithmetic methods, with op string:
         f(op, name, str_rep, default_axis=None, fill_zeros=None, **eval_kwargs)
-    comp_method : function, optional,
+    comp_method : function (optional)
         factory for rich comparison - signature: f(op, name, str_rep)
+    bool_method : function (optional)
+        factory for boolean methods - signature: f(op, name, str_rep)
     use_numexpr : bool, default True
         whether to accelerate with numexpr, defaults to True
     force : bool, default False
@@ -234,9 +236,16 @@ def f(self, other):
              __isub__=_wrap_inplace_method(new_methods["__sub__"]),
              __imul__=_wrap_inplace_method(new_methods["__mul__"]),
              __itruediv__=_wrap_inplace_method(new_methods["__truediv__"]),
-             __ipow__=_wrap_inplace_method(new_methods["__pow__"]), ))
+             __ifloordiv__=_wrap_inplace_method(new_methods["__floordiv__"]),
+             __imod__=_wrap_inplace_method(new_methods["__mod__"]),
+             __ipow__=_wrap_inplace_method(new_methods["__pow__"])))
     if not compat.PY3:
-        new_methods["__idiv__"] = new_methods["__div__"]
+        new_methods["__idiv__"] = _wrap_inplace_method(new_methods["__div__"])
+    if bool_method:
+        new_methods.update(
+            dict(__iand__=_wrap_inplace_method(new_methods["__and__"]),
+                 __ior__=_wrap_inplace_method(new_methods["__or__"]),
+                 __ixor__=_wrap_inplace_method(new_methods["__xor__"])))
 
     add_methods(cls, new_methods=new_methods, force=force, select=select,
                 exclude=exclude)

pandas/tests/frame/test_operators.py

@@ -1161,6 +1161,33 @@ def test_inplace_ops_identity(self):
         assert_frame_equal(df2, expected)
         assert df._data is df2._data
 
+        # no id change
+        df = DataFrame({'a': [1., 2., 3.]})
+        expected = id(df)
+        df += 2.
+        assert id(df) == expected
+        df -= 2.
+        assert id(df) == expected
+        df *= 2.
+        assert id(df) == expected
+        df /= 2.
+        assert id(df) == expected
+        df **= 2.
+        assert id(df) == expected
+        df //= 2.
+        assert id(df) == expected
+        df %= 2.
+        assert id(df) == expected
+
+        df = DataFrame({'a': [1, 0, 5]})
+        expected = id(df)
+        df &= 1
+        assert id(df) == expected
+        df |= 1
+        assert id(df) == expected
+        df ^= 1
+        assert id(df) == expected
+
     def test_alignment_non_pandas(self):
         index = ['A', 'B', 'C']
         columns = ['X', 'Y', 'Z']