Add skipna keyword into astype

Author: makbigc

pandas/core/generic.py

@@ -5776,7 +5776,7 @@ def _to_dict_of_blocks(self, copy=True):
             for k, v, in self._data.to_dict(copy=copy).items()
         }
 
-    def astype(self, dtype, copy=True, errors="raise"):
+    def astype(self, dtype, copy=True, errors="raise", skipna=True):
         """
         Cast a pandas object to a specified dtype ``dtype``.
 
@@ -5798,6 +5798,9 @@ def astype(self, dtype, copy=True, errors="raise"):
             - ``ignore`` : suppress exceptions. On error return original object.
 
             .. versionadded:: 0.20.0
+        skipna : bool, default True
+            Exclude NA/null values in type conversion.
+
 
         Returns
         -------
@@ -5884,7 +5887,7 @@ def astype(self, dtype, copy=True, errors="raise"):
                         "the key in Series dtype mappings."
                     )
                 new_type = dtype[self.name]
-                return self.astype(new_type, copy, errors)
+                return self.astype(new_type, copy, errors, skipna)
 
             for col_name in dtype.keys():
                 if col_name not in self:
@@ -5896,7 +5899,12 @@ def astype(self, dtype, copy=True, errors="raise"):
             for col_name, col in self.items():
                 if col_name in dtype:
                     results.append(
-                        col.astype(dtype=dtype[col_name], copy=copy, errors=errors)
+                        col.astype(
+                            dtype=dtype[col_name],
+                            copy=copy,
+                            errors=errors,
+                            skipna=skipna
+                        )
                     )
                 else:
                     results.append(results.append(col.copy() if copy else col))
@@ -5911,7 +5919,9 @@ def astype(self, dtype, copy=True, errors="raise"):
 
         else:
             # else, only a single dtype is given
-            new_data = self._data.astype(dtype=dtype, copy=copy, errors=errors)
+            new_data = self._data.astype(
+                dtype=dtype, copy=copy, errors=errors, skipna=skipna
+            )
             return self._constructor(new_data).__finalize__(self)
 
         # GH 19920: retain column metadata after concat

pandas/core/internals/blocks.py

@@ -536,10 +536,10 @@ def f(mask, val, idx):
 
         return self.split_and_operate(None, f, False)
 
-    def astype(self, dtype, copy=False, errors="raise", **kwargs):
-        return self._astype(dtype, copy=copy, errors=errors, **kwargs)
+    def astype(self, dtype, copy=False, errors="raise", skipna=True, **kwargs):
+        return self._astype(dtype, copy=copy, errors=errors, skipna=skipna, **kwargs)
 
-    def _astype(self, dtype, copy=False, errors="raise", **kwargs):
+    def _astype(self, dtype, copy=False, errors="raise", skipna=True, **kwargs):
         """Coerce to the new type
 
         Parameters
@@ -606,7 +606,7 @@ def _astype(self, dtype, copy=False, errors="raise", **kwargs):
             else:
                 values = self.get_values(dtype=dtype)
 
-            skipna = kwargs.pop("skipna", True)
+            #skipna = kwargs.pop("skipna", True)
             # _astype_nansafe works fine with 1-d only
             vals1d = values.ravel()
             try: