minor rec2excel enhancements

jdh2358 · jdh2358 · commit 5d6994e63320 · 2007-12-05T19:27:14.000Z
svn path=/trunk/matplotlib/; revision=4629
diff --git a/examples/loadrec.py b/examples/loadrec.py
@@ -9,4 +9,6 @@
 ax = fig.add_subplot(111)
 ax.plot(a.date, a.adj_close, '-')
 fig.autofmt_xdate()
+
+mlab.rec2excel(a, 'test.xls', colnum=4)
 show()
diff --git a/lib/matplotlib/mlab.py b/lib/matplotlib/mlab.py
@@ -2367,7 +2367,7 @@ def xlformat_factory(format):
             xlstyle.num_format_str = '#,##;[RED]-#,##'
         elif isinstance(format, FormatPercent):
             zeros = ''.join(['0']*format.precision)
-            xlstyle.num_format_str = '0.%s%;[RED]-0.%s%'%(zeros, zeros)
+            xlstyle.num_format_str = '0.%s%%;[RED]-0.%s%%'%(zeros, zeros)
             format.scale = 1.
         else:
             xlstyle = None
@@ -2376,12 +2376,14 @@ def xlformat_factory(format):
 
         return format
 
-    def rec2excel(r, ws, formatd=None, rownum=0):
+    def rec2excel(r, ws, formatd=None, rownum=0, colnum=0):
         """
         save record array r to excel pyExcelerator worksheet ws
         starting at rownum.  if ws is string like, assume it is a
         filename and save to it
 
+        start writing at rownum, colnum
+
         formatd is a dictionary mapping dtype name -> FormatXL instances
 
         The next rownum after writing is returned
@@ -2399,14 +2401,20 @@ def rec2excel(r, ws, formatd=None, rownum=0):
             formatd = dict()
 
         formats = []
+        font = excel.Font()
+        font.bold = True
+
+        stylehdr = excel.XFStyle()
+        stylehdr.font = font
+
         for i, name in enumerate(r.dtype.names):
             dt = r.dtype[name]
             format = formatd.get(name)
             if format is None:
                 format = defaultformatd.get(dt.type, FormatObj())
 
             format = xlformat_factory(format)
-            ws.write(rownum, i, name)
+            ws.write(rownum, colnum+i, name, stylehdr)
             formats.append(format)
 
         rownum+=1
@@ -2419,12 +2427,12 @@ def rec2excel(r, ws, formatd=None, rownum=0):
                 format = formats[i]
                 val = format.toval(val)
                 if format.xlstyle is None:
-                    ws.write(rownum, i, val)
+                    ws.write(rownum, colnum+i, val)
                 else:
                     if safe_isnan(val):
-                        ws.write(rownum, i, 'NaN')
+                        ws.write(rownum, colnum+i, 'NaN')
                     else:
-                        ws.write(rownum, i, val, format.xlstyle)
+                        ws.write(rownum, colnum+i, val, format.xlstyle)
             rownum += 1
 
         if autosave:
