Fix Values.append

synw · Jun 16, 2019 · 07521d0 · 07521d0
1 parent f18750b
commit 07521d0
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/dataswim/data/transform/values.py b/dataswim/data/transform/values.py
@@ -52,8 +52,12 @@ def append(self, vals: list, index=None):
         :example: ``ds.append([0, 2, 2, 3, 4])``
         """
         try:
-            # self.df.append(vals)
-            self.df.append(pd.DataFrame(columns=self.df.columns, data=[vals]))
+            if index is not None:
+                self.df = self.df.append(pd.DataFrame(columns=self.df.columns,
+                                                      data=[vals], index=[index]))
+            else:
+                self.df = self.df.append(pd.DataFrame(columns=self.df.columns,
+                                                      data=[vals]))
         except Exception as e:
             self.err(e, self.append, "Can not append row")
             return