Checking split dict size.

1 files changed, 14 insertions, 11 deletions

diff --git a/src/argaze/DataStructures.py b/src/argaze/DataStructures.py
index 7fb8cd8..fcd91e1 100644
--- a/src/argaze/DataStructures.py
+++ b/src/argaze/DataStructures.py
@@ -232,24 +232,27 @@ class TimeStampedBuffer(collections.OrderedDict):
         df.drop(exclude, inplace=True, axis=True)
 
         # Split columns
-        splited_columns = []
-        for column in df.columns:
+        if len(split) > 0:
 
-            if column in split.keys():
+            splited_columns = []
+            
+            for column in df.columns:
+
+                if column in split.keys():
 
-                df[split[column]] = pandas.DataFrame(df[column].tolist(), index=df.index)
-                df.drop(column, inplace=True, axis=True)
+                    df[split[column]] = pandas.DataFrame(df[column].tolist(), index=df.index)
+                    df.drop(column, inplace=True, axis=True)
 
-                for new_column in split[column]:
+                    for new_column in split[column]:
 
-                    splited_columns.append(new_column)
+                        splited_columns.append(new_column)
 
-            else:
+                else:
 
-                splited_columns.append(column)
+                    splited_columns.append(column)
 
-        # Reorder splited columns
-        df = df[splited_columns]
+            # Reorder splited columns
+            df = df[splited_columns]
 
         # Append timestamps as index column
         df['timestamp'] = self.keys()