Hi,

I noticed numpy.loadtxt has support for gzipped text files, but not for
bz2'd files.  Here's a 3 line patch to add bzip2 support to loadtxt.

Ryan

-- 
Ryan May
Graduate Research Assistant
School of Meteorology
University of Oklahoma
Index: numpy/lib/io.py
===================================================================
--- numpy/lib/io.py     (revision 5953)
+++ numpy/lib/io.py     (working copy)
@@ -320,6 +320,9 @@
         if fname.endswith('.gz'):
             import gzip
             fh = gzip.open(fname)
+        elif fname.endswith('.bz2'):
+            import bz2
+            fh = bz2.BZ2File(fname)
         else:
             fh = file(fname)
     elif hasattr(fname, 'seek'):
_______________________________________________
Numpy-discussion mailing list
Numpy-discussion@scipy.org
http://projects.scipy.org/mailman/listinfo/numpy-discussion

Reply via email to