kuba/dotfiles
1
0
Fork 0
Code Issues Pull Requests Projects Releases Activity

Read npy files

Browse Source
This commit is contained in:
Jakub Fojt
2024-01-18 16:40:44 +01:00
parent af4e2d4364
commit 75b08a39f6
1 changed files with 11 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
.modules.kuba/.tools.sh
View File

@@ -11,12 +11,18 @@ except ImportError:
class Wrap: class Wrap:
def __init__(self, fname): def __init__(self, fname):
n = np.load(fname, allow_pickle=True) n = np.load(fname, allow_pickle=True)
files = n.files
globs = globals() globs = globals()
if isinstance(n, np.lib.npyio.NpzFile):
files = n.files
print('Contains files', files) print('Contains files', files)
for f in files: for f in files:
setattr(self, f, n[f]) setattr(self, f, n[f])
globs[f] = n[f] globs[f] = n[f]
else:
print('Loaded data of shape', n.shape)
setattr(self, 'data', n)
globs['data'] = n
argv = argv[argv.index('--')+1:] argv = argv[argv.index('--')+1:]
archives = [Wrap(arg) for arg in argv] archives = [Wrap(arg) for arg in argv]