Skip to content

Commit

Permalink
csvSaver now saves .csv files correctly
Browse files Browse the repository at this point in the history
  • Loading branch information
mfrlin committed Jul 23, 2013
1 parent 9471f1c commit 8657b2b
Showing 1 changed file with 11 additions and 7 deletions.
18 changes: 11 additions & 7 deletions Orange/data/io.py
Expand Up @@ -214,15 +214,19 @@ def constr_vars(inds):

def csvSaver(filename, data, delimiter='\t'):
with open(filename, 'w') as csvfile:
flags = ['']*len(data.domain)
class_var = data.domain.class_var
if class_var:
flags[data.domain.indices[class_var.name]] = 'class'

writer = csv.writer(csvfile, delimiter=delimiter)
writer.writerow([d.name for d in data.domain]) # write attribute names
writer.writerow([str(d.var_type).lower() for d in data.domain]) # write attribute types
writer.writerow(flags) # write flags
if delimiter == '\t':
flags = ['']*len(data.domain)
class_var = data.domain.class_var
metas = data.doman.metas
if class_var:
flags[data.domain.indices[class_var.name]] = 'class'
if metas:
for m in metas:
flags[data.domain.indices[m.name]] = 'm'
writer.writerow([str(d.var_type).lower() for d in data.domain]) # write attribute types
writer.writerow(flags) # write flags
for ex in data: # write examples
writer.writerow(ex)

Expand Down

0 comments on commit 8657b2b

Please sign in to comment.