cdfdist.py

   1 import sys
   2
   3 def main(argv=None):
   4     if not argv:
   5         argv = sys.argv
   6
   7     if len(argv) < 4:
   8         print "usage: python %s bins percent infile" % sys.argv[0]
   9         sys.exit(1)
  10
  11     bins = int(argv[1])
  12     percent = int(argv[2])
  13     infilename = argv[3]
  14
  15     printCdfDist(bins, percent, infilename)
  16
  17
  18 def printCdfDist(bins, percent, infilename):
  19     print cdfDist(bins, percent, infilename)
  20
  21
  22 def cdfDist(bins, percent, infilename):
  23     infile = open(infilename)
  24     binsList = [0] * bins
  25
  26     for line in infile:
  27         fields = line.strip().split()
  28         index = 0
  29         for binCdf in fields[-1 * bins:]:
  30             if int(binCdf) > percent:
  31                 binsList[index] += 1
  32                 break
  33
  34             index += 1
  35
  36     infile.close()
  37     return binsList
  38
  39
  40 if __name__ == "__main__":
  41     main(sys.argv)