handle list with nan

Author: sanggon6107

pandas/core/generic.py

@@ -28,7 +28,6 @@
 
 from pandas._libs import lib
 from pandas._libs.lib import is_range_indexer
-from pandas._libs.missing import NA
 from pandas._libs.tslibs import (
     Period,
     Timestamp,
@@ -114,6 +113,7 @@
     is_bool_dtype,
     is_dict_like,
     is_extension_array_dtype,
+    is_float_dtype,
     is_list_like,
     is_number,
     is_numeric_dtype,
@@ -9713,6 +9713,7 @@ def _where(
         if axis is not None:
             axis = self._get_axis_number(axis)
 
+        has_nan: bool = False
         # align the cond to same shape as myself
         cond = common.apply_if_callable(cond, self)
         if isinstance(cond, NDFrame):
@@ -9728,9 +9729,13 @@ def _where(
         else:
             if not hasattr(cond, "shape"):
                 cond = np.asanyarray(cond)
+            if is_float_dtype(cond) and np.isnan(cond).any():
+                has_nan = True
             if cond.shape != self.shape:
                 raise ValueError("Array conditional must be same shape as self")
             cond = self._constructor(cond, **self._construct_axes_dict(), copy=False)
+            if has_nan:
+                cond = cond.replace({0.0: False, 1.0: True})
             cond = cond.fillna(True)
 
         # make sure we are boolean
@@ -10113,13 +10118,13 @@ def mask(
         # see gh-21891
         if not hasattr(cond, "__invert__"):
             cond = np.array(cond)
+            if is_float_dtype(cond) and np.isnan(cond).any():
+                cond = cond.astype(bool)
 
         if isinstance(cond, np.ndarray):
-            if all(x is NA or lib.is_bool(x) or x is np.nan for x in cond.flatten()):
-                if not cond.flags.writeable:
-                    cond.setflags(write=True)
-                cond[isna(cond)] = False
-                cond = cond.astype(bool)
+            if not cond.flags.writeable:
+                cond.setflags(write=True)
+            cond[isna(cond)] = False
 
         return self._where(
             ~cond,