colsum.py

   1 import sys
   2
   3 def main(argv=None):
   4     if not argv:
   5         argv = sys.argv
   6
   7     print "colsum: version 1.3"
   8     if len(argv) < 3:
   9         print "usage: python %s field filename" % argv[0]
  10         print "\n\tfields are counted starting at zero.\n"
  11         sys.exit(1)
  12
  13     fieldID = int(argv[1])
  14     filename = argv[2]
  15
  16     count = colsum(fieldID, filename)
  17     print count
  18
  19
  20 def colsum(fieldID, filename):
  21     infile = open(filename)
  22     count = 0
  23
  24     for line in infile:
  25         fields = line.strip().split()
  26         try:
  27             if "." in fields[fieldID]:
  28                 count += float(fields[fieldID])
  29             else:
  30                 count += int(fields[fieldID])
  31         except ValueError:
  32             pass
  33
  34     infile.close()
  35     return count
  36
  37
  38 if __name__ == "__main__":
  39     main(sys.argv)