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
