BUG: Fix DataFrame.apply(..., raw=True) not calling with raw array

Author: kernc

pandas/core/apply.py

@@ -179,7 +179,7 @@ def get_result(self):
             return self.apply_empty_result()
 
         # raw
-        elif self.raw and not self.obj._is_mixed_type:
+        elif self.raw:
             return self.apply_raw()
 
         return self.apply_standard()

pandas/tests/frame/test_apply.py

@@ -235,7 +235,7 @@ def test_apply_broadcast_error(self, int_frame_const_col):
         with pytest.raises(ValueError):
             df.apply(lambda x: Series([1, 2]), axis=1, result_type="broadcast")
 
-    def test_apply_raw(self, float_frame):
+    def test_apply_raw(self, float_frame, mixed_type_frame):
         result0 = float_frame.apply(np.mean, raw=True)
         result1 = float_frame.apply(np.mean, axis=1, raw=True)
 
@@ -250,6 +250,13 @@ def test_apply_raw(self, float_frame):
         expected = float_frame * 2
         tm.assert_frame_equal(result, expected)
 
+        # Mixed dtype
+        def _assert_raw(x):
+            assert isinstance(x, np.ndarray)
+
+        mixed_type_frame.apply(_assert_raw, raw=True)
+        mixed_type_frame.apply(_assert_raw, axis=1, raw=True)
+
     def test_apply_axis1(self, float_frame):
         d = float_frame.index[0]
         tapplied = float_frame.apply(np.mean, axis=1)