BF: manual fix for csv writing in example

matthew-brett · matthew-brett · commit 9a4a038d1805 · 2013-01-26T20:14:20.000-08:00
Use manual fix for Python 2 / 3 differences for clarity in example.
diff --git a/examples/labs/write_paradigm_file.py b/examples/labs/write_paradigm_file.py
@@ -7,6 +7,7 @@
 
 See Pinel et al., BMC neuroscience 2007 for reference
 """
+import sys
 import csv
 
 import numpy as np
@@ -45,7 +46,11 @@
 sess = np.zeros(np.size(time)).astype('int8')
 pdata = np.vstack((sess, cid, time)).T
 csvfile = 'localizer_paradigm.csv'
-fid = open(csvfile, "wt")
+# Opening files for CSV writing differs between Python 2 and 3
+if sys.version_info[0] >= 3: # Python 3
+    fid = open(csvfile, "w", newline = '')
+else: # Python 2
+    fid = open(csvfile, "wb")
 writer = csv.writer(fid, delimiter=' ')
 for row in pdata:
     writer.writerow(row)
