Merge pull request #68 from fact-project/copy_columns

Copy columns
fact-project · May 29, 2017 · 58e9bdb · 58e9bdb
2 parents 486a9ed + 916430e
commit 58e9bdb
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/fact/io.py b/fact/io.py
@@ -84,6 +84,8 @@ def read_h5py(file_path, key='data', columns=None, mode='r+'):
         # get all columns of which don't have more than one value per row
         if columns is None:
             columns = [col for col in group.keys() if group[col].ndim == 1]
+        else:
+            columns = copy(columns)
 
         df = pd.DataFrame()
         for col in columns:
@@ -133,6 +135,8 @@ def read_h5py_chunked(file_path, key='data', columns=None, chunksize=None, mode=
         # get all columns of which don't have more than one value per row
         if columns is None:
             columns = [col for col in group.keys() if group[col].ndim == 1]
+        else:
+            columns = copy(columns)
 
         n_rows = h5py_get_n_rows(file_path, key=key)