def convert_wmap2fh(wmapfn, outfn): ff = tarfile.open(wmapfn) d = {} for fn1 in ff.getnames(): if fn1=="description.txt": continue buf = ff.extractfile(fn1) print "Extracting %s....."%fn1 d[fn1]=np.loadtxt(buf,dtype='f4')[:,1] ff.close() ND.write_file(outfn, d)
def convert_wmap2fh(wmapfn, outfn): ff = tarfile.open(wmapfn) d = {} for fn1 in ff.getnames(): if fn1 == "description.txt": continue buf = ff.extractfile(fn1) print "Extracting %s....." % fn1 d[fn1] = np.loadtxt(buf, dtype='f4')[:, 1] ff.close() ND.write_file(outfn, d)
def convert_cosmo2fh(chainname, chainnum, outname): colnames=parse_colnames(chainname) ncols = len(colnames) fn = chainname+"_%d.txt"%chainnum data = np.loadtxt(fn,dtype='f4') if data.shape[1]!=ncols: print "Expected %d columns, got %d columns..."%(ncols, data.shape[1]) raise RuntimeError, "Incorrect number of columns found" d = {} for ii,icol in enumerate(colnames): d[icol] = data[:,ii] ND.write_file(outname,d)
from __future__ import print_function from ndfilehandler import read_file, write_file import numpy as np ## Write out a file data = {} data['farr'] = np.arange(10, dtype='f8') data['iarr'] = np.arange(27, dtype='i4').reshape((3, 3, 3)) write_file("ndtest", data) ## Read in the file data2 = read_file("ndtest", None) print(data['farr'] == data2['farr']) print(data['iarr'] == data2['iarr']) print(data2['iarr'].shape)
from ndfilehandler import read_file, write_file; import numpy as np ## Write out a file data = {} data['farr'] = np.arange(10,dtype='f8') data['iarr'] = np.arange(27,dtype='i4').reshape((3,3,3)) write_file("ndtest",data) ## Read in the file data2 = read_file("ndtest",None) print data['farr']==data2['farr'] print data['iarr']==data2['iarr'] print data2['iarr'].shape