Merge pull request #17 from Kirill-Lekhov/master

Simplified the replace_null_objects function in the pandasplus module
m1kc · Jun 1, 2022 · 65f5cc6 · 65f5cc6
2 parents 6b1438e + 0477adc
commit 65f5cc6
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/notalib/pandas/pandasplus.py b/notalib/pandas/pandasplus.py
@@ -1,4 +1,4 @@
-from pandas import DataFrame, NaT, notnull
+from pandas import DataFrame, notnull
 
 
 def row_to_dict(d, key_as=None):
@@ -32,7 +32,6 @@ def replace_null_objects(dataframe: DataFrame, new_value=None) -> DataFrame:
 	it to None, I recommend using this function.
 	"""
 
-	dataframe = dataframe.replace({NaT: new_value})
 	dataframe = dataframe.where(notnull(dataframe), new_value)
 
 	return dataframe