JoeGermuska · September 14, 2009 21:01 · Sep 14, 2009 · Sep 14, 2009 · Sep 14, 2009 · Sep 14, 2009
diff --git a/csvcut b/csvcut
@@ -32,7 +32,7 @@ delimiter = ','
 cols = [0, ]
 if opts:
     opts = dict(opts)
-    show_headers = bool('-h' in opts or '--header' in opts)
+    show_headers = '-h' in opts
     if '-f' in opts:
         cols = opts['-f'].split(",")
     if '-d' in opts:

diff --git a/csvcut b/csvcut
@@ -6,6 +6,8 @@ Note that fields are zero-based, as opposed to 'cut' where they are 1-based.
 
 Leveraged from/motivated by an example from @bycoffe
 
+Should use something better than getopt, but this works...
+
 Usage: 
     csvcut foobar.csv
     (prints the first column of each row of foobar.csv)
@@ -15,9 +17,13 @@ Usage:
 
     csvcut -f 0,2 -d "|" foobar.csv
     (prints the first and third columns of the pipe-delimited foobar.csv)
+
+    csvcut -h foobar.csv
+    (prints the values of the first line of foobar.csv, preceded by the field index which would 
+     be used to display that column.  If present, the -f option will be ignored.)
 """
 import sys, csv, getopt
-opts, args = getopt.getopt(sys.argv[1:], "f:d:", ["fields=", "delimiter="])
+opts, args = getopt.getopt(sys.argv[1:], "f:d:h", [])
 if args:
     i = open(args[0])
 else:
@@ -26,11 +32,17 @@ delimiter = ','
 cols = [0, ]
 if opts:
     opts = dict(opts)
+    show_headers = bool('-h' in opts or '--header' in opts)
     if '-f' in opts:
         cols = opts['-f'].split(",")
     if '-d' in opts:
         delimiter = opts['-d']
 for row in csv.reader(i, delimiter=delimiter):
+    if show_headers:
+        for i,c in enumerate(row):
+            print "%3i: %s" % (i,c)
+        break    
     for c in cols:
         print row[int(c)],
-    print
+    print
+
diff --git a/csvcut b/csvcut
@@ -22,8 +22,8 @@ if args:
     i = open(args[0])
 else:
     i = sys.stdin
-delimiter = ',' 
-cols = [1]
+delimiter = ','
+cols = [0, ]
 if opts:
     opts = dict(opts)
     if '-f' in opts:

diff --git a/csvcut b/csvcut
@@ -1,9 +1,6 @@
 #!/usr/bin/env python
 """
 Like cut, but for CSVs. To be used from a shell command line.
- 
-Change row[1] to the row index to be printed. row[1] will print the second
-item in the row.
 
 Note that fields are zero-based, as opposed to 'cut' where they are 1-based.
 
@@ -15,19 +12,25 @@ Usage:
     
     head -10 foobar.csv | csvcut -f 0,2
     (prints the first and third columns of the first ten lines of foobar.csv)
+
+    csvcut -f 0,2 -d "|" foobar.csv
+    (prints the first and third columns of the pipe-delimited foobar.csv)
 """
 import sys, csv, getopt
-opts, args = getopt.getopt(sys.argv[1:], "f:", ["fields="])
+opts, args = getopt.getopt(sys.argv[1:], "f:d:", ["fields=", "delimiter="])
 if args:
     i = open(args[0])
 else:
     i = sys.stdin
+delimiter = ',' 
+cols = [1]
 if opts:
-    cols = opts[0][1].split(",")    
-else:
-    cols = [0]    
-for row in csv.reader(i):
+    opts = dict(opts)
+    if '-f' in opts:
+        cols = opts['-f'].split(",")
+    if '-d' in opts:
+        delimiter = opts['-d']
+for row in csv.reader(i, delimiter=delimiter):
     for c in cols:
         print row[int(c)],
-    print
-
+    print
diff --git a/csvcut b/csvcut
@@ -5,7 +5,16 @@ Like cut, but for CSVs. To be used from a shell command line.
 Change row[1] to the row index to be printed. row[1] will print the second
 item in the row.
 
-Leveraged from an example from @bycoffe
+Note that fields are zero-based, as opposed to 'cut' where they are 1-based.
+
+Leveraged from/motivated by an example from @bycoffe
+
+Usage: 
+    csvcut foobar.csv
+    (prints the first column of each row of foobar.csv)
+    
+    head -10 foobar.csv | csvcut -f 0,2
+    (prints the first and third columns of the first ten lines of foobar.csv)
 """
 import sys, csv, getopt
 opts, args = getopt.getopt(sys.argv[1:], "f:", ["fields="])
@@ -16,7 +25,7 @@ else:
 if opts:
     cols = opts[0][1].split(",")    
 else:
-    cols = [1]    
+    cols = [0]    
 for row in csv.reader(i):
     for c in cols:
         print row[int(c)],

diff --git a/csvcut b/csvcut
@@ -0,0 +1,24 @@
+#!/usr/bin/env python
+"""
+Like cut, but for CSVs. To be used from a shell command line.
+ 
+Change row[1] to the row index to be printed. row[1] will print the second
+item in the row.
+
+Leveraged from an example from @bycoffe
+"""
+import sys, csv, getopt
+opts, args = getopt.getopt(sys.argv[1:], "f:", ["fields="])
+if args:
+    i = open(args[0])
+else:
+    i = sys.stdin
+if opts:
+    cols = opts[0][1].split(",")    
+else:
+    cols = [1]    
+for row in csv.reader(i):
+    for c in cols:
+        print row[int(c)],
+    print
+
No results found