use fillna instead of replace

wq · Oct 18, 2014 · 35c767d · 35c767d
1 parent 9880bef
commit 35c767d
Showing 1 changed file with 1 addition and 5 deletions.
diff --git a/rest_pandas/serializers.py b/rest_pandas/serializers.py
@@ -18,11 +18,7 @@ def get_dataframe(self, data):
         if index:
             if self.index_none_value is not None:
                 for key in index:
-                    dataframe.replace(
-                        {key: None},
-                        {key: self.index_none_value},
-                        inplace=True,
-                    )
+                    dataframe[key].fillna(self.index_none_value, inplace=True)
             dataframe.set_index(index, inplace=True)
         else:
             # Name auto-index column to ensure valid CSV output