Try to keep the original column types

bra-fsn · Apr 11, 2022 · ead74de · ead74de
1 parent 21c1917
commit ead74de
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/hashable_df/__init__.py b/hashable_df/__init__.py
@@ -13,12 +13,15 @@ def f(value):
 
 def hashable_df(df):
     """Return a DataFrame with hashable cell values."""
-    return df.applymap(f)
+    # applymap might change types. Try to restore the original.
+    orig_types = df.dtypes
+    df = df.applymap(f).astype(orig_types, errors="ignore")
+    return df
 
 
 def hashable_series(ser):
     """Return a Series with hashable values."""
     return ser.map(f)
 
 
-__version__ = '0.0.6'
+__version__ = '0.0.7'