diff options
Diffstat (limited to 'util.py')
-rw-r--r-- | util.py | 5 |
1 files changed, 4 insertions, 1 deletions
@@ -1,8 +1,11 @@ | |||
1 | import numpy as np | 1 | import numpy as np |
2 | 2 | ||
3 | def load_csv_file(fname, *args, **kargs): | 3 | def load_csv_file(fname, *args, **kargs): |
4 | return np.genfromtxt(open(fname), delimiter=",", comments="#", | 4 | f = open(fname) |
5 | data = np.genfromtxt(f, delimiter=",", comments="#", | ||
5 | *args, **kargs) | 6 | *args, **kargs) |
7 | f.close() # don't leak file handles | ||
8 | return data | ||
6 | 9 | ||
7 | def select(keep, rows): | 10 | def select(keep, rows): |
8 | ok_rows = np.zeros(len(rows), dtype=bool) | 11 | ok_rows = np.zeros(len(rows), dtype=bool) |