BUG: Properly handle lists for .mask

Closes pandas-devgh-21891.

Author: gfyoung

doc/source/whatsnew/v0.24.0.txt

@@ -155,7 +155,7 @@ For situations where you need an ``ndarray`` of ``Interval`` objects, use
 :meth:`numpy.asarray` or ``idx.astype(object)``.
 
 .. ipython:: python
-  
+
    np.asarray(idx)
    idx.values.astype(object)
 
@@ -482,6 +482,7 @@ Reshaping
 - Bug in :func:`pandas.concat` when joining resampled DataFrames with timezone aware index (:issue:`13783`)
 - Bug in :meth:`Series.combine_first` with ``datetime64[ns, tz]`` dtype which would return tz-naive result (:issue:`21469`)
 - Bug in :meth:`Series.where` and :meth:`DataFrame.where` with ``datetime64[ns, tz]`` dtype (:issue:`21546`)
+- Bug in :meth:`Series.mask` and :meth:`DataFrame.mask` with ``list`` conditionals (:issue:`21891`)
 -
 -
 

pandas/core/generic.py

@@ -7941,6 +7941,10 @@ def mask(self, cond, other=np.nan, inplace=False, axis=None, level=None,
         inplace = validate_bool_kwarg(inplace, 'inplace')
         cond = com._apply_if_callable(cond, self)
 
+        # see gh-21891
+        if not hasattr(cond, "__invert__"):
+            cond = np.array(cond)
+
         return self.where(~cond, other=other, inplace=inplace, axis=axis,
                           level=level, try_cast=try_cast,
                           errors=errors)

pandas/tests/frame/test_indexing.py

@@ -3010,6 +3010,14 @@ def test_mask_callable(self):
         tm.assert_frame_equal(result,
                               (df + 2).mask((df + 2) > 8, (df + 2) + 10))
 
+    def test_mask_list(self):
+        # see gh-21891
+        df = DataFrame([1, 2])
+        res = df.mask([[True], [False]])
+
+        exp = DataFrame([np.nan, 2])
+        tm.assert_frame_equal(res, exp)
+
     def test_head_tail(self):
         assert_frame_equal(self.frame.head(), self.frame[:5])
         assert_frame_equal(self.frame.tail(), self.frame[-5:])

pandas/tests/series/indexing/test_boolean.py

@@ -582,6 +582,15 @@ def test_where_dt_tz_values(tz_naive_fixture):
     assert_series_equal(exp, result)
 
 
+def test_where_list():
+    # see gh-21891
+    s = Series([1, 2])
+    res = s.mask([True, False])
+
+    exp = Series([np.nan, 2])
+    tm.assert_series_equal(res, exp)
+
+
 def test_mask():
     # compare with tested results in test_where
     s = Series(np.random.randn(5))