BUG: Add fillna so that cond doesnt contain NA at the beginning of _w…

…here. (pandas-dev#60729)
sanggon6107 · Jan 23, 2025 · bc9a942 · bc9a942
1 parent 60325b8
commit bc9a942
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/pandas/core/generic.py b/pandas/core/generic.py
@@ -9674,6 +9674,12 @@ def _where(
         if axis is not None:
             axis = self._get_axis_number(axis)
 
+        # We should not be filling NA. See GH#60729
+        if isinstance(cond, np.ndarray):
+            cond[np.isnan(cond)] = True
+        elif isinstance(cond, NDFrame):
+            cond = cond.fillna(True)
+
         # align the cond to same shape as myself
         cond = common.apply_if_callable(cond, self)
         if isinstance(cond, NDFrame):