readings-06.py

   1 import sys
   2 import numpy as np
   3
   4 def main():
   5     script = sys.argv[0]
   6     action = sys.argv[1]
   7     filenames = sys.argv[2:]
   8     assert action in ['--min', '--mean', '--max'], \
   9            'Action is not one of --min, --mean, or --max: ' + action
  10     if len(filenames) == 0:
  11         process(sys.stdin, action)
  12     else:
  13         for f in filenames:
  14             process(f, action)
  15
  16 def process(filename, action):
  17     data = np.loadtxt(filename, delimiter=',')
  18
  19     if action == '--min':
  20         values = data.min(axis=1)
  21     elif action == '--mean':
  22         values = data.mean(axis=1)
  23     elif action == '--max':
  24         values = data.max(axis=1)
  25
  26     for m in values:
  27         print m
  28
  29 main()