def save_csv(filename, dat, **kw):
'''
save_csv(filename, d) writes a pandas dataframe d to a CSV file with the given name. If pandas
cannot be loaded, then an error is raised. If d is not a dataframe, to_dataframe() is called
on it.
All optional arguments are passed along to the pandas.DataFrame.to_csv function.
'''
import pandas
from neuropythy.util import to_dataframe
d = to_dataframe(dat)
if any(filename.endswith(s) for s in ('.gz', '.bz2', '.lzma')):
with gzip.open(filename, 'wt', newlines='') as fl: d.to_csv(fl, **kw)
else:
with open(filename, 'wt', newlines='') as fl: d.to_csv(fl, **kw)
return data
def save_tsv(filename, dat, sep='\t', **kw):
'''
save_tsv(filename, d) writes a pandas dataframe d to a TSV file with the given name. If pandas
cannot be loaded, then an error is raised. If d is not a dataframe, to_dataframe() is called
on it.
All optional arguments are passed along to the pandas.DataFrame.to_csv function. Note that this
function is identical to save_csv() except that it has a default sep value of '\t' instead of
','.
'''
import pandas
from neuropythy.util import to_dataframe
d = to_dataframe(dat)
if any(filename.endswith(s) for s in ('.gz', '.bz2', '.lzma')):
with gzip.open(filename, 'wt', newlines='') as fl: d.to_csv(fl, sep=sep, **kw)
else:
with open(filename, 'wt', newlines='') as fl: d.to_csv(fl, sep=sep, **kw)
return data
