novice/python/readings-05.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     for f in filenames:
  11         process(f, action)
  12
  13 def process(filename, action):
  14     data = np.loadtxt(filename, delimiter=',')
  15
  16     if action == '--min':
  17         values = data.min(axis=1)
  18     elif action == '--mean':
  19         values = data.mean(axis=1)
  20     elif action == '--max':
  21         values = data.max(axis=1)
  22
  23     for m in values:
  24         print m
  25
  26 main()